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) {int totalSum=0;int size=mat.length;for(int i=0;i<size;++i){int reverseIndex=size-i-1;totalSum +=mat[i][i];if (i!=reverseIndex){totalSum+=mat[i][reverseIndex];}}return totalSum;}
}
因为是方正所以size=mat.length;
主对角线:mat[i][i];
副对角线:mat[i][size-i-1];
如果是偶数列(行)则不会有重复,如果是奇数列,则但行标=列标时,中间元素会重复,不能加。