[C++知识库]LeetCode每日一题 - 矩阵对角线元素的和

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

C语言

int diagonalSum(int** mat, int matSize, int* matColSize)
{
    int i, sum=0;
    int m = matSize-1;
    for(i=0;i<matSize;i++)
        sum += mat[i][i];
    for(i=0;i<matSize;i++)
    {
        if(i == m-i)
            continue;
        sum += mat[i][m-i];
    }
    return sum;
}

执行用时12ms，内存消耗6.4MB。

Python3

class Solution:
    def diagonalSum(self, mat: List[List[int]]) -> int:
        res=0
        m=len(mat)
        for i in range(m):
            res+=mat[i][i]
        for j in range(m):
            if j == m-j-1:
                continue
            res += mat[j][m-1-j]
        return res

执行用时28ms，内存消耗15.2MB。