1572. 矩阵对角线元素的和
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int length=mat.size(); //获取正方体边数
int sum=0;//创建和的参数
if(length%2==0){//为偶数
for(int i=0;i<length;i++){
sum+=mat[i][i];
sum+=mat[i][length-i-1];
}
}else{
for(int i=0;i<length;i++){
sum+=mat[i][i];
if(i!=length-i-1){
sum+=mat[i][length-i-1];
}
}
}
return sum;
}
};