1572. 矩阵对角线元素的和
给你一个正方形矩阵
mat
,请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
用每行的行标分别表示在每一行上的主副对角线的元素。
$ mat[i][i],mat[i][cnt-1-i]$
但需要注意的是,主对角线和副对角线可能会有交点,这个时候就需要判断是否是相交的,即判定列坐标是否相同。
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int cnt=mat.size(),ans=0;
for (int i=0;i<cnt;i++)
ans+=i==cnt-1-i?mat[i][i]:mat[i][i]+mat[i][cnt-1-i];
return ans;
}
};