Java解决矩阵对角线元素的和问题
01 题目
给你一个正方形矩阵 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].length1 <= n <= 1001 <= mat[i][j] <= 100
02 知识点
- 二维数组
03 我的题解
java
public class shuzu04 {
public static void main(String[] args) {
int[][] mat=new int[][] {{1,2,3},
{4,5,6},
{7,8,9}};
System.out.println(diagonalSum(mat));
}
public static int diagonalSum(int[][] mat) {
int m=mat[0].length;
int count=0;
for (int i = 0; i < m; i++) {
count+=mat[i][i];
count+=mat[i][m-1-i];
}
if (m%2==1) {
int n=(m-1)/2;
count-=mat[n][n];
}
return count;
}
}