leetcode 1572. Matrix Diagonal Sum（矩阵对角线之和）

计算矩阵对角线元素之和,

最新推荐文章于 2025-12-03 23:35:37 发布

原创最新推荐文章于 2025-12-03 23:35:37 发布 · 150 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #矩阵 #算法

该代码片段提供了一个方法来计算矩阵主对角线和第二对角线元素的和，避免了重复添加同一元素。通过按行遍历并同时考虑两端的元素，提高了效率。

在这里插入图片描述

求矩阵主对角线和第二对角线上的元素之和，同一元素不重复加。

思路：

可以取mat[ i ][ i ]相加。但是取二维数组元素要比取一维数组元素效率低。
所以按行（一维数组）取从两端到中间的元素。即mat[ i ]和mat[ n - 1 - i ].

public int diagonalSum(int[][] mat) {
    int n = mat.length;
    if(n == 1) return mat[0][0];
    int sum = 0;
    int i = 0;

    for(int[] row : mat) {
        if(n-1-i == i) sum += row[i];
        else sum = sum + row[i] + row[n-1-i];
        i ++;
    }
    return sum;
}