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

总结

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

相关推荐
宝耶27 分钟前
面试常问问题:Java基础篇
java·面试·职场和发展
Wendy_robot29 分钟前
【滑动窗口+哈希表/数组记录】Leetcode 438. 找到字符串中所有字母异位词
c++·算法·leetcode
程序员-King.35 分钟前
day49—双指针+贪心—验证回文串(LeetCode-680)
算法·leetcode·贪心算法·双指针
躲在云朵里`1 小时前
IDEA搭建环境的五种方式
java·ide·intellij-idea
喵手1 小时前
从 Java 到 Kotlin:在现有项目中迁移的最佳实践!
java·python·kotlin
阑梦清川1 小时前
AI超级智能体项目教程(二)---后端项目初始化(设计knif4j接口文档的使用)
java·前端·数据库
转基因1 小时前
Codeforces Round 1020 (Div. 3)(题解ABCDEF)
数据结构·c++·算法
jack xu11 小时前
高频面试题:如何保证数据库和es数据一致性
java·大数据·数据库·mysql·elasticsearch
幽络源小助理2 小时前
微信小程序文章管理系统开发实现
java·微信小程序·springboot
jzshmyt2 小时前
大内存生产环境tomcat-jvm配置实践
java·jvm·tomcat