python-leecode-020矩阵对角元素的和

020矩阵对角元素的和

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
**示例  1：**
![示例1](https://img-blog.csdnimg.cn/20210418204613236.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80Mzk4ODEzMQ==,size_16,color_FFFFFF,t_70)
输入：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

解法1: 循环 相加
class Solution:
    def diagonalSum(self, mat: List[List[int]]) -> int:
        n = 0
        row = len(mat)
        for i in range(row):
            if row - i - 1 == i:
                n += mat[i][i]
                continue
            n += mat[i][i] + mat[row - i - 1][i]
        return n
       
解法2: 非对角线元素置零
import numpy as np
class Solution:
    def diagonalSum(self, mat: List[List[int]]) -> int:
        mat = np.array(mat)
        for i in range(mat.shape[0]):
            for j in range(mat.shape[1]):
                if i != j and i + j + 1 != mat.shape[0]:
                    mat[i, j] = 0
        return int(np.sum(mat))