LC.P1572[矩阵对角线元素的和]

方法一：模拟

1
2
3
4
5
6
7
8
9
10
class Solution {
    public int diagonalSum(int[][] mat) {
        int n = mat.length, ans = 0;
        for (int i = 0, j = n - 1; i < n; ++i, --j) {
            ans += mat[i][i] + mat[i][j];
            if (i == j) ans -= mat[i][i];
        }
        return ans;
    }
}
  • 时间复杂度：$O(n)$
  • 空间复杂度：$O(1)$