Problem: 1572. 矩阵对角线元素的和

超简单思路-矩阵对角线元素和

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