主對角線矩陣的總和JAVA

Question

我需要在Java中找到矩陣中主對角線的總和。我的程序適用於矩陣（如1x2和2x2），但是當行數大於列數時（即3x2），我的代碼會拋出一個超出界限的異常。有什麼方法可以解決它嗎？

// sum of major diagonal in matrix 
public static int majorDiagonal(int[][] array){ 
    int sumMD = 0; 
    for (int i = 0, j = 0, k = array.length; i < k && j < k; i++, j++) { 
     sumMD += array[i][j]; 
    } 
return sumMD; 
} 

Answer 1

你並不需要同時i和j，因爲他們總是在你的代碼相同的值。

因此，您的延續條件i < k && j < k沒有太大意義，因爲如果i小於k，那麼j也會如此。

將for循環的延續條件更改爲i < k && i < array[0].length。這將檢查i是否小於列數和行數，這是您需要的條件。

for (int i = 0, k = array.length; i < k && i < array[0].length; i++) { 
    sumMD += array[i][i]; 
}