给你一个正方形矩阵 mat
,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
示例 1:
输入:mat = [[1,2,3], [4,5,6], [7,8,9]] 输出:25 解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25 请注意,元素 mat[1][1] = 5 只会被计算一次。
示例 2:
输入:mat = [[1,1,1,1], [1,1,1,1], [1,1,1,1], [1,1,1,1]] 输出:8
示例 3:
输入:mat = [[5]] 输出:5
提示:
n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100
class Solution {
public int diagonalSum(int[][] mat) {
int sum = 0;
for (int i = 0,j = 0,m = mat.length-1; i < mat.length; i++,j++,m--) {
sum += mat[i][j] + mat[i][m];//计算对角线之和
}
if(mat.length % 2 != 0) {//奇数减去中间重复的
return sum - mat[(mat.length-1)/2][(mat.length-1)/2];
}else {
return sum;
}
}
}