1. 程式人生 > >求矩陣的兩對角線上的元素之和

求矩陣的兩對角線上的元素之和

/*題目描述
求矩陣的兩對角線上的元素之和
輸入
矩陣的行數N
和一個N*N的整數矩陣a[N]N
輸出
所輸矩陣的兩對角線上的元素之和
樣例輸入
3
1 2 3
4 5 6
7 8 9
樣例輸出
25*/

import java.util.*;

public class Text_1138 {

public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    int n = sc.nextInt();
    int sum = 0;
    int[][] a = new int[n][n];
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            a[i][j] = sc.nextInt();
            if (i == j || i+j+1==n) {//對角線的索引規律
                sum = sum + a[i][j];
            }
        }
    }
    System.out.println(sum);
}
}