1572. 矩阵对角线元素的和
2023年8月12日19:07:51
简单
给你一个正方形矩阵 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].length1 <= n <= 1001 <= mat[i][j] <= 100
答案
class Solution {
public int diagonalSum(int[][] mat) {
int c = mat.length;
int r = mat[0].length;
int sum = 0;
for(int i = 0,j = 0,x=c-1;i<c&&j<r&&x>=0;i++,j++,x--){
if(i != x){
sum += mat[i][j] + mat[x][j];
}else{
sum += mat[i][j];
}
}
return sum;
}
}