每日一題——矩陣對角線元素和

給你一個正方形矩陣 mat，請你返回矩陣對角線元素的和。

請你返回在矩陣主對角線上的元素和副對角線上且不在主對角線上元素的和。

（若是階數爲奇數，對角線交點的元素只會被加一次）

 

今日題目太腦殘

個人答案：

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];
    }
};