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

原创于 2023-05-08 18:30:00 发布 · 128 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #矩阵 #算法

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

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

思路：

可以取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;
}