其他分享
首页 > 其他分享> > LeetCode #1572. Matrix Diagonal Sum

LeetCode #1572. Matrix Diagonal Sum

作者:互联网

题目

1572. Matrix Diagonal Sum


解题方法

遍历数组每次累加mat[i][j]和mat[i][len(mat)-1-j],如果j == len(mat) - 1 - j的话就去掉一个mat[i][j],最后返回Sum。
时间复杂度:O(mn)
空间复杂度:O(1)


代码

class Solution:
    def diagonalSum(self, mat: List[List[int]]) -> int:
        Sum = 0
        i = j = 0
        while i < len(mat):
            Sum += mat[i][j] + mat[i][len(mat)-1-j]
            if j == len(mat) - 1 - j:
                Sum -= mat[i][j]
            i += 1
            j += 1
        return Sum

标签:Matrix,int,Sum,Diagonal,len,1572,复杂度,mat
来源: https://www.cnblogs.com/RatsCommander/p/14074576.html