题目:
给你一个正方形矩阵 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
C语言
int diagonalSum(int** mat, int matSize, int* matColSize)
{
int i, sum=0;
int m = matSize-1;
for(i=0;i<matSize;i++)
sum += mat[i][i];
for(i=0;i<matSize;i++)
{
if(i == m-i)
continue;
sum += mat[i][m-i];
}
return sum;
}
执行用时12ms,内存消耗6.4MB。
Python3
class Solution:
def diagonalSum(self, mat: List[List[int]]) -> int:
res=0
m=len(mat)
for i in range(m):
res+=mat[i][i]
for j in range(m):
if j == m-j-1:
continue
res += mat[j][m-1-j]
return res
执行用时28ms,内存消耗15.2MB。