1572. 矩阵对角线元素的和

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

代码如下:

class Solution {

    /**
     * @param Integer[][] $mat
     * @return Integer
     */
    function diagonalSum($mat) {
        $col = count($mat[0]);
        $row = count($mat);
        $newArr = [];
        for ($i=0; $i < $row; $i++) {
            for ($j=0; $j < $col; $j++) {
                if ($i == $j || ($i + $j == $row - 1) ) {
                    $newArr[] = $mat[$i][$j];
                }
            }
        }
        return array_sum($newArr);
    }
}

本文链接:https://itarvin.com/detail-133.aspx

登录或者注册以便发表评论

登录

注册