6-2 求矩阵主对角线元素的和 （pta java 第六章）

编写一个程序，求n x n的double类型矩阵中主对角线上所有数字的和。

输入格式:

第一行输入矩阵的阶数n（0<=n<=10）。 第二行输入矩阵的n x n个double类型的数值。

输出格式:

输出n x n的double类型矩阵中对角线上所有数字的和（double型数值）。

输入样例:

4
1 2 3 4 5 6.5 7 8 9 10 11 12 13 14 15 16

输出样例:

34.5

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        double sum =0 ;
        double array[][] = new double[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                array[i][j] = input.nextDouble();
                if (i==j){
                    sum = sum + array[i][j];
                }
            }
        }
        System.out.println(sum);
    }
}