【LeetCode】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 sum = 0;

        for (int i = 0; i < mat.length; i++) {
            if (i == mat[0].length - 1 - i) {
                sum += mat[i][i];
            } else {
                sum += mat[i][i];
                sum += mat[i][mat[0].length - 1 - i];
            }
        }

        return sum;
    }
}

总结

观察矩阵的对角线,发现通常情况每一行都有两个元素需要加入和的运算,且这两个元素的索引都与当前行的索引有一定的数学规律。那么将每行的这两个元素都加入和,若这两个元素索引重合,说明这行只有一个元素需要加入。

相关推荐
Lizhihao_5 分钟前
JAVA-堆 和 堆排序
java·开发语言
极客先躯10 分钟前
高级java每日一道面试题-2025年3月21日-微服务篇[Nacos篇]-什么是Nacos?
java·开发语言·微服务
南玖yy12 分钟前
数据结构C语言练习(栈)
c语言·数据结构·算法
工业互联网专业18 分钟前
基于springboot+vue的动漫交流与推荐平台
java·vue.js·spring boot·毕业设计·源码·课程设计·动漫交流与推荐平台
雷渊21 分钟前
深入分析Spring的事务隔离级别及实现原理
java·后端·面试
阿镇吃橙子25 分钟前
一些手写及业务场景处理问题汇总
前端·算法·面试
酱酱哥玩AI29 分钟前
Trae编译器:实现多目标班翠鸟优化算法(IPKO)无人机路径规划仿真(Python版),完整代码
算法
rebel32 分钟前
Java获取excel附件并解析解决方案
java·后端
MPCTHU42 分钟前
二叉树、排序算法与结构图
数据结构·算法·排序算法
并不会43 分钟前
多线程案例-单例模式
java·学习·单例模式·单线程·多线程·重要知识