题意:
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
数据范围:
n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100
解法:
直接模拟即可:
for(int i=0;i<n;i++){
ans+=a[i][i];
if(n-i-1!=i){
ans+=a[i][n-i-1];
}
}
code:
class Solution {
public:
int diagonalSum(vector<vector<int>>& a) {
int n=a.size();
int ans=0;
for(int i=0;i<n;i++){
ans+=a[i][i];
if(n-i-1!=i){
ans+=a[i][n-i-1];
}
}
return ans;
}
};