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)。