给你一个正方形矩阵 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
思路
首先,获取矩阵的大小size,然后定义一个变量sum,初始化为0,用来累计对角线元素的和。
然后,通过两个循环遍历矩阵的主对角线和次对角线上的元素。对于主对角线,元素的行列坐标相同,即为(i, i),所以直接将对应元素加到sum上。对于次对角线,元素的行列坐标之和为size-1,即为(i, size - i - 1),所以也将对应元素加到sum上。需要注意的是,当矩阵的大小为奇数时,主对角线和次对角线会有一个交点,这个元素会被重复计算,所以在计算次对角线元素和时,需要跳过这个元素。
最后,返回sum,即矩阵对角线元素的和。
AC代码
/*
* @lc app=leetcode.cn id=1572 lang=cpp
*
* [1572] 矩阵对角线元素的和
*/
// @lc code=start
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int size = mat.size();
long long sum = 0;
for (int i = 0; i < size; i++) {
sum += mat[i][i];
}
for (int i = 0; i < size; i++) {
int j = size - i - 1;
if (i == j) {
continue;
}
sum += mat[i][j];
}
return sum;
}
};
// @lc code=end