LeetCode练习十九

最新推荐文章于 2024-10-17 14:59:31 发布

m0_74384965

最新推荐文章于 2024-10-17 14:59:31 发布

阅读量211

点赞数 3

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_74384965/article/details/140751196

版权

1572.矩阵对角线元素的和

给你一个正方形矩阵 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

示例 3：

输入：mat = [[5]]
输出：5

提示：

n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100

class Solution:
    def diagonalSum(self, mat: List[List[int]]) -> int:
        total = 0
        n = len(mat)
        if n == 1:
            return mat[0][0]
        else:
            for i in range(n):
                total += mat[i][i] + mat[i][n - i - 1]
            if n % 2 == 1:
                total -= mat[int(n/2)][int(n/2)]
                return total
            else:
                return total