leetcode_1572 矩阵对角线元素的和

85 阅读1分钟

要求

给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

示例  1:

image.png

输入: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:
        res = 0
        for i in range(len(mat)):
            res += mat[i][i]
            res += mat[i][len(mat) - 1 - i]
        if len(mat) % 2 != 0:
            res -= mat[len(mat)//2][len(mat) // 2]
        return res

image.png

解题思路:循环遍历矩阵的每一行,加上mat[i][i],加上mat[i][len(mat)-1-i],如果矩阵行数为奇数,减去最中心元素,比较简单。