//
给你一个正方形矩阵 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
//
//
👍 50👎 0
package leetcode.editor.cn;
// 1572:矩阵对角线元素的和
public class MatrixDiagonalSum {
public static void main(String[] args) {
Solution solution = new MatrixDiagonalSum().new Solution();
// TO TEST
}
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int diagonalSum(int[][] mat) {
int sum = 0;
int size = mat.length;
for (int i = 0; i < size; i++) {
sum += mat[i][i] + mat[size - 1 - i][i];
}
return size % 2 == 0 ? sum : sum - mat[size / 2][size / 2];
}
}
//leetcode submit region end(Prohibit modification and deletion)
}