矩阵对角线元素的和
问题:
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
思路:
例:
输入:
mat = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
输出:25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25
- 将对角线上的元素(主、副对角线)加起来。
- 如果矩阵行数为奇数,则减去中心元素,为偶数则不减。
- 返回结果。
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);
}
};