一个不知错因的题目

编写一个程序，求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 [] agrs){
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();
        if(n==0){
            System.out.println(0);
        }
        else{
            double [][] b=new double[n] [n];
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                b[i][j]=in.nextDouble();
            }
        }
        double sum=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(i==j){
                    sum=sum+b[i][j];
                }
            }
        }
        System.out.println(sum);
        }
        
    }
}
 

很简单的一道题目

不知道为什么提交的时候总是部分正确

对于0的判断是我后来加上去的 但还是不对

真的头疼有木有大佬给讲一下部分正确是少考虑了什么情况

！！！！！！就剩这一个题交不上这就很难受