本文主要是介绍1572.矩阵对角线元素的和(Java),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目描述:
给你一个正方形矩阵 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;}
}
这篇关于1572.矩阵对角线元素的和(Java)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!