内容思路参考力扣官方题解
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
也就是求对角线的和,不包括中间的重复元素。
方法一:
常规思路,直接相加
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int n=mat.size();
int sum=0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if((i==j)||(i+j==n-1)){
sum+=mat[i][j];
}
}
}
return sum;
}
};
这里直接遍历 i 和 j ,一旦有满足条件的就直接相加,不存在有重复元素的情况
时间复杂度:O(n^2) 空间复杂度:O(1)
方法二:
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int n=mat.size();
int sum=0;
int mid=n/2;
for(int i=0;i<n;i++){
sum+=mat[i][i]+mat[i][n-1-i];
}
return sum-mat[mid][mid]*(n&1);
}
};
如果 n 是奇数的话,最中间的格子会被加入两次。所以 n 为奇数的时候,我们需要减掉矩阵最中心的那个值。这里通过n&1实现奇偶性判断。因为奇数的二进制末位一定是1,1&1=1,则为奇,0&1=0则为偶。
时间复杂度:O(n)
空间复杂度:O(1)