浅谈 对角线元素的和 问题

矩阵对角线元素的和

问题:
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

思路:
例:
输入:

mat = [[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]]

输出:25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25

  1. 将对角线上的元素(主、副对角线)加起来。
  2. 如果矩阵行数为奇数,则减去中心元素,为偶数则不减。
  3. 返回结果。
class Solution {
public:
    int diagonalSum(const vector<vector<int>>& mat) {
        auto len = mat.size();
        auto sum = 0;
        for(auto i = 0; i < len; ++i) 
            sum += mat[i][i] + mat[i][len - i - 1];
        auto j = len >> 1;
        return sum - mat[j][j] * (len & 1);
    }
};
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 像素格子 设计师:CSDN官方博客 返回首页