难度简单
给你一个正方形矩阵 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
遍历
c++
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int sum=0;
for(int i=0;i<mat.size();i++)
{
for (int j=0;j<mat[0].size();j++)
{
if (i==j) sum+=mat[i][j];
else if (i+j==mat.size()-1) sum+=mat[i][j];
}
}
return sum;
}
};
python
class Solution:
def diagonalSum(self, mat: List[List[int]]) -> int:
sum=0
for i in range (len(mat)):
for j in range (len(mat[0])):
if i==j:
sum+=mat[i][j]
elif i+j==len(mat)-1:
sum+=mat[i][j]
return sum