java 求5*5矩阵对角线之和

假设变量array是一个5*5的二维数组.

double sum = 0;
for(int i=0; i<5; i++){
for(int j=0; j<5; j++){
if(i=j || i+j=4){
sum += array[i][j];
}
}
}
System.out.println(sum);

这样求出来的是矩阵两条对角线上数字的总和。