7-5 求矩阵主对角线元素的和 (20 分)

7-5 求矩阵主对角线元素的和 (20 分)

编写一个程序，求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.*;

class Main{
    public static void main(String[] args) {
       Scanner cin = new Scanner(System.in);
       int n = cin.nextInt();
       double [][]d = new double[n][n];
      double sum = 0;

        for(int i = 0; i < n; i ++ ){
            for( int  j = 0; j < n; j ++ ){
                d[i][j] = cin.nextDouble();
                if( i == j ){
                    sum += d[i][j];
                }
            }
        }

//      for(int i = 0; i < n; i ++ ){
//              sum += d[i][i];
//      }

        System.out.println(sum);

       cin.close();
    }
}