Java解决矩阵对角线元素的和问题

Java解决矩阵对角线元素的和问题

01 题目

给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

示例 1:

img

输入: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

02 知识点

  • 二维数组

03 我的题解

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;
    }
}

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>