給你一個正方形矩陣 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]; } };
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/173092.html
標籤:C++
上一篇:學習第52天
