题目:
给你一个正方形矩阵 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
来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1、题目说是正方形矩阵,即行数等于列数,必然存在对角线,且有两种情况,对角线是否有重合元素。
2、两条对角线是对称的,所以每行两个对角线元素的下标存在一定关系,即mat[i][i] 与mat[i][n-i-1]
3、当i=n-i-1时,即对角线有重合的元素,只需要累计一次。
代码:
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int sum=0;
int n=mat.size();
for(int i=0;i<n;i++)
{
sum+=mat[i][i];
if(i!=n-i-1) //每行对角线元素不是重合元素时
sum+=mat[i][n-i-1];
}
return sum;
}
};