题目描述:
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
输入:
mat = [[1,2,3],
[4,5,6],
[7,8,9]]
输出:
25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25
请注意,元素 mat[1][1] = 5 只会被计算一次。
代码实现:
public class Main{public static void main(String[] args) {//案例int[][] mat = new int[][]{{7, 3, 1, 9}, {3, 4, 6, 9}, {6, 9, 6, 6}, {9, 5, 8, 5}};System.out.println(diagonalSum(mat));//55}public static int diagonalSum(int[][] mat) {int len = mat.length;//获取正方形的边长int sum = 0;//求和//获取主对角线元素之和for (int i = 0; i < mat.length; i++) {sum += mat[i][i];}//获取副对角线元素之和for (int i = mat.length - 1; i >= 0; i--) {sum += mat[i][mat.length - 1 - i];}//判断奇偶if (len % 2 != 0) {//如果是奇数,要减去中心的重复元素sum = sum - mat[len / 2][len / 2];}return sum;}
}