1572. 矩阵对角线元素的和

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

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

示例 1:

复制代码
输入: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
java 复制代码
class Solution {
    public int diagonalSum(int[][] mat) {
        int totalSum=0;
        int size=mat.length;
        for(int i=0;i<size;++i){
            int reverseIndex=size-i-1;
            totalSum +=mat[i][i];
            if (i!=reverseIndex){
                totalSum+=mat[i][reverseIndex];
            }
        }

        return totalSum;
    }
}

因为是方正所以size=mat.length;

主对角线:mat[i][i];

副对角线:mat[i][size-i-1];

如果是偶数列(行)则不会有重复,如果是奇数列,则但行标=列标时,中间元素会重复,不能加。

相关推荐
前端炒粉4 小时前
18.矩阵置零(原地算法)
javascript·线性代数·算法·矩阵
AI Chen4 小时前
【矩阵分析与应用】【第1章 矩阵与线性方程组】【1.6.2.2 迹的循环置换性质】
矩阵·
暴风鱼划水5 小时前
三维重建【0-D】3D Gaussian Splatting:相机标定原理与步骤
算法·3d
大千AI助手6 小时前
HOSVD(高阶奇异值分解):高维数据的“解剖术”
人工智能·线性代数·矩阵·张量·svd·hosvd·高阶奇异值分解
mount_myj7 小时前
敏感信息屏蔽(一)【java】
java·算法·极课堂
先做个垃圾出来………7 小时前
偏移量解释
数据结构·算法
FanXing_zl7 小时前
基于整数MCU的FOC控制定标策略深度解析
单片机·嵌入式硬件·mcu·算法·定点运算·q15
立志成为大牛的小牛8 小时前
数据结构——三十三、Dijkstra算法(王道408)
数据结构·笔记·学习·考研·算法·图论
地平线开发者9 小时前
mul 与 reduce_sum 的优化实例
算法·自动驾驶