力扣Leetcode第1572题，矩阵对角线元素的和的Python解法

Description

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

Codes

class Solution(object):
    def diagonalSum(self, mat):
        """
        :type mat: List[List[int]]
        :rtype: int
        """
        count = 0

        len1 = len(mat)
        for i in range(len1):
            if len1%2 == 0:
                count += mat[i][i] + mat[i][len1-i-1]
            else:
                if i == len1 / 2:
                    count += mat[i][i]
                else:
                    count += mat[i][i] + mat[i][len1-i-1]
        
        return count