0
点赞
收藏
分享

微信扫一扫

C++——list的简要介绍

灵魂跑者 2023-08-16 阅读 55

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

在这里插入图片描述

输入:mat = [[1,2,3],
            [4,5,6],
            [7,8,9]]
输出:25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25
请注意,元素 mat[1][1] = 5 只会被计算一次。

示例二

输入:mat = [[1,1,1,1],
            [1,1,1,1],
            [1,1,1,1],
            [1,1,1,1]]
输出:8

示例三

输入:mat = [[5]]
输出:5

代码实现

public class DiagonalSum {
    public static int diagonalSum(int[][] mat) {
        int sum = 0;
        int len = mat.length - 1;
        for (int i = 0; i < mat.length; i++) {
            if (i != len) {
                sum += mat[i][i] + mat[i][len];
            } else {
                sum += mat[i][i];
            }
            len --;
        }
        return sum;
    }
    public static void main(String[] args) {
        int[][] arr = {
                {1, 2, 3},
                {4, 5, 6},
                {7, 8, 9}
        };
        System.out.println(diagonalSum(arr));
    }
}
举报

相关推荐

0 条评论