矩阵对角线元素的和

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

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

class Solution {
public:
    int diagonalSum(vector<vector<int>>& mat) {
        // 感觉自己好蠢啊
        int ans = 0;
        int r = mat.size();
        int c = mat[0].size();
        for(int i = 0; i < r; ++i){
            ans += mat[i][i];
        }
        for(int i = 0; i < r; ++i){
            if(r-i-1 != i){
                ans += mat[i][r-i-1];
            }
        }
        return ans;
    }
};

多观察，多思考，找规律，不难的。