给你一个正方形矩阵 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
public class Solution1572 {
public int diagonalSum(int[][] mat) {
int out = 0;
for (int i = 0, j = 0; i < mat.length; i++, j++) {
// System.out.println(mat[i][j]);
out = out + mat[i][j];
}
for (int i = 0, j = mat.length - 1; i < mat.length; i++, j--) {
if (i != j) {
// System.out.println(mat[i][j]);
out = out + mat[i][j];
}
}
return out;
}
public static void main(String[] args) {
Solution1572 s = new Solution1572();
int[][] mat = { { 1, 2, 3 }, { 4, 5, 6 }, { 7, 8, 9 } };
System.out.println(s.diagonalSum(mat));
}
}