题目链接
题意:给定一个
n
∗
n
n*n
n∗n矩阵求主副对角线元素的和;
思路:两重for循环即可
AC代码:
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int n=mat.size();
int sum=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(i==j||i+j==n-1)
sum+=mat[i][j];
return sum;
}
};