给你一个正方形矩阵 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
java代码:
class Solution {
public int diagonalSum(int[][] mat) {
int x = 0; //主对角线相加和
int y = 0; //副对角线上且不在主对角线上元素的和
//循环相加即可
for (int i = 0; i < mat.length; i++) {
x += mat[i][i];
//如果到相交的点,y不相加,进入下次循环
if(i == mat.length - 1 - i){
continue;
}
y += mat[i][mat.length - 1 - i];
}
return x + y;
}
}