1572.矩阵对角线元素的和(Java)

题目描述:

给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

1572.矩阵对角线元素的和(Java)_第1张图片

输入:

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

你可能感兴趣的:(力扣刷题,java,矩阵,开发语言)