Matrix Diagonal Sum 矩陣對角線元素的和

問題描述：

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

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

$$\begin{gathered} n==mat.length==mat[i].length \\ 1<=n<=100 \\ 1<=mat[i][j]<=100 \end{gathered}$$

分析

這個問題就是矩陣的對角線遍歷。

• 主對角線元素的坐標一定是$a[i][i]$,
• 副對角線的坐標就是$a[i][j],i+j==n?1$

其次，要求重合的位置僅計算一次，而只有當n為odd的情況下，才會重合。

時間復雜度 $O(N)$,空間復雜度 $O(1)$

代碼

Math

public int diagonalSum(int[][] mat) {int sum = 0,n = mat.length,mid = n/2;if(n==1)return mat[0][0];for(int i = 0;i<n;i++){sum += mat[i][i]+ mat[i][n-1-i];            }return (n&1)==0?sum:sum - mat[mid][mid];}

時間復雜度$O(N)$

空間復雜度$O(1)$

Tag

Matrix