每日一题——LeetCode1572.矩阵对角线元素的和

zbsnzj2024-03-03 14:48

方法一遍历矩阵

如果矩阵中某个位置（x,y）处于对角线上，那么这个位置必定满足：

x=y 或 x+y = len-1 （len为矩阵长度）

javascript 复制代码

var diagonalSum = function(mat) {
    let len = mat.length;
    let sum = 0;
    for (let i = 0; i < len; i++) {
        for (let j = 0; j < len; j++) {
            if (i == j || i + j == len - 1) {
                sum += mat[i][j];
            }
        }
    }
    return sum;
};

消耗时间和内存情况：

方法二枚举对角线元素：

逐行遍历，记当前行号为i，每一行的对角线元素为(i,i)和(i,len-i-1)

如果len为奇数则对角线交点会被计算两次，需要减掉一次

用len&1，如果len为奇数结果为1，len为偶数结果为0

javascript 复制代码

var diagonalSum = function(mat) {
    let len = mat.length, mid = Math.floor(len / 2);
    let sum = 0;
    for (let i = 0; i < len; ++i) {
        sum += mat[i][i] + mat[i][len - 1 - i];
    }
    return sum - mat[mid][mid] * (len & 1);
};

消耗时间和内存情况：