LeetCode 1572. 矩阵对角线元素的和(模拟)

题意:

给你一个正方形矩阵 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;
    }
};

你可能感兴趣的:(LeetCode 1572. 矩阵对角线元素的和(模拟))