1572. 矩阵对角线元素的和

Posted on | In
Words count in article: 216 | Reading time ≈ 1

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

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

示例1：

1
2
3
4
5
6
输入：mat = [[1,2,3],
            [4,5,6],
            [7,8,9]]
输出：25
解释：对角线的和为：1 + 5 + 9 + 3 + 7 = 25
请注意，元素 mat[1][1] = 5 只会被计算一次。

示例2:

1
2
3
4
5
输入：mat = [[1,1,1,1],
            [1,1,1,1],
            [1,1,1,1],
            [1,1,1,1]]
输出：8

示例3：

1
2
输入：mat = [[5]]
输出：5

提示：

  • n == mat.length == mat[i].length
  • 1 <= n <= 100
  • 1 <= mat[i][j] <= 100

解法

解法一：

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
public int diagonalSum(int[][] mat) {
        int sum = 0;

        for (int i = 0;i < mat.length;i++) {
            sum += mat[i][i];
            sum += mat[i][mat.length - i - 1];
        }

        if (1 == mat.length % 2) {
            sum -= mat[mat.length / 2][mat.length / 2];
        }
        return sum;
    }
0%