给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
//我们知道矩阵中某个位置 (i,j) 处于对角线上,则一定满足下列条件之一:
//i=j;
//i+j=n−1;
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int n=mat.size();
int ans=0;
for(int i=0;i<n;i++){
int m=mat[i].size();
for(int j=0;j<m;j++){
if(i==j||i+j==n-1){
ans+=mat[i][j];
}
}
}
return ans;
}
};

被折叠的 条评论
为什么被折叠?



