20.9.6 周赛 5491. 矩阵对角线元素的和 简单

题目

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

示例  1：

输入：mat = [[1,2,3],
  [4,5,6],
  [7,8,9]]
输出：25
解释：对角线的和为：1 + 5 + 9 + 3 + 7 = 25
请注意，元素 mat[1][1] = 5 只会被计算一次。
示例  2：

输入：mat = [[1,1,1,1],
  [1,1,1,1],
  [1,1,1,1],
  [1,1,1,1]]
输出：8
示例 3：

输入：mat = [[5]]
输出：5

提示：

n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100

思路

1. 除去中间会重复的，每一行都有两个数要加起来，一个是对角线元素，另一个刚好是长度-对角线

代码

class Solution {
public:
    int diagonalSum(vector<vector<int>>& mat) {
        int res=0;
        for(int i=0;i<mat.size();i++){
            res+=mat[i][i];
            res+=mat[i][mat[0].size()-i-1];
            if(i==mat[0].size()-i-1) res-=mat[i][i];
        }
        return res;
    }
};