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++){
            for(int j = 0 ; j < mat[i].length ; j++){
                if(i==j){
                sum += mat[i][j];
                break;
                }
            }
            for(int j = mat[i].length-1 ; j >= 0 ; j--){
                if(i+j==mat.length-1){
                    sum += mat[i][j];
                    break;
                }
            }
        }
        if(mat.length%2!=0){
        sum-=mat[(mat.length-1)/2][(mat.length-1)/2];
        }
        return sum;
    }
}

方法二:枚举对角线元素:

java 复制代码
class Solution {
    public int diagonalSum(int[][] mat) {
        /*
        如果记行号为i,则当前下处于对角线的元素为:(i,i)和(i,n-i-1)
        并且当n为奇数,最中间交叉处的值会被加两次,所以得减掉一次。
         */
         int  n = mat.length , sum = 0;
         for(int i = 0 ; i < n ; i ++){
             sum+=mat[i][i] + mat[i][n-i-1];
         }
         if(n%2!=0) return sum-mat[n/2][n/2];//判断奇偶性还可以用n&1
         return sum;
    }
}

PS: n&1为什么能够判断奇偶,是因为在二进制中,奇数的最低位一定是1,偶数的最低位一定是0。所以无论其他位是什么,取决奇偶性的在于最低位。1&1=1,故为奇数;0&1=0,故为偶数。

相关推荐
戊辰happy12 分钟前
arcface
算法
陌上花开࿈1 小时前
调用第三方接口
java
浊酒南街1 小时前
决策树python实现代码1
python·算法·决策树
Aileen_0v01 小时前
【玩转OCR | 腾讯云智能结构化OCR在图像增强与发票识别中的应用实践】
android·java·人工智能·云计算·ocr·腾讯云·玩转腾讯云ocr
冠位观测者3 小时前
【Leetcode 热题 100】208. 实现 Trie (前缀树)
数据结构·算法·leetcode
桂月二二3 小时前
Java与容器化:如何使用Docker和Kubernetes优化Java应用的部署
java·docker·kubernetes
liuxin334455663 小时前
学籍管理系统:实现教育管理现代化
java·开发语言·前端·数据库·安全
小马爱打代码4 小时前
设计模式详解(建造者模式)
java·设计模式·建造者模式
小王爱吃月亮糖4 小时前
C++的23种设计模式
开发语言·c++·qt·算法·设计模式·ecmascript
栗子~~4 小时前
idea 8年使用整理
java·ide·intellij-idea