LeetCode 1572. 矩阵对角线元素的和

LeetCode 1572. 矩阵对角线元素的和

描述

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

示例 3：

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

提示：

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

题解

class Solution {
    public int diagonalSum(int[][] mat) {
        // 给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。
        // 请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
        int n = mat.length;
        // 1..n
        // n..1
        // n为奇数 n/2+1 为中心重复
        // n为偶数 不重复
        int res = 0;
        int j =0;
        int k =n-1;
        for(int[] row:mat){
            // i 行
            res+=row[j];
            res+=row[k];
            k--;
            j++;
        }
        if(n%2==0){
            // 是偶数
            return res;

        }else{
            // 是奇数
           return res - mat[n/2][n/2];
        }

    }
}