超简单思路-矩阵对角线元素和
1.思路
把前几种矩阵对角线元素相加的情况列举一下,方可发现主对角线与副对角线元素下标的规律。
2.解题方法
因为是正方形矩阵,所以len(mat) == len(mat[0]),用这个作为循环条件,把所有元素加起来,如果len(mat)%2 == 1,那么就再把对角线相交的那个点的值再减一遍即可。
3.复杂度
时间复杂度: O(n)
4.Code
class Solution:
def diagonalSum(self, mat: List[List[int]]) -> int:
final_num=0
len_mat=len(mat)
for i in range(len_mat):
final_num+=mat[i][i]
final_num+=mat[i][len_mat-i-1]
if len(mat)%2:
final_num-=mat[len_mat//2][len_mat//2]
return final_num
# 2 00 11 01 10
# 3 00 11 22 02 11 20
# 4 00 11 22 33 03 12 21 30