每日一题——矩阵对角线元素和

给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。spa

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。code

(若是阶数为奇数,对角线交点的元素只会被加一次)blog

 

今日题目太脑残io

个人答案:class

class Solution {
public:
    int diagonalSum(vector<vector<int>>& mat) {
        int res=0,size=mat.size(),mid=size>>1;       
        for(int i=0;i<size;++i)
        {
            res+=mat[i][i];
            res+=mat[i][size-1-i];
        }
        return size%2==0?res:res-mat[mid][mid];
    }
};
相关文章
相关标签/搜索