1572. 矩阵对角线元素的和 - 力扣（LeetCode）

题目

给你一个正方形矩阵 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 ans = 0, n = mat.size()-1;for(int i = 0; i <= n; ++i) {// 主对角线ans += mat[i][i];// 副对角线if(i != n-i) ans += mat[i][n-i];}return ans;}
};

复杂度分析

• 时间复杂度：O(n)。
• 空间复杂度：O(1)。