求一个n阶方阵对角线元素之和。

import java.util.*;
public class Test4 {
	public static void main(String args[]) {
		System.out.print("请输入阶数n：");
		Scanner input = new Scanner(System.in);
		int n=input.nextInt();
		int num[][]= new int[n][n];
		
		System.out.println("请输入"+n+"阶方阵");
		int sum1=0,sum2=0;  //存两对角线的和
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++)
			{
				num[i][j] = input.nextInt();
				if(i==j)
					sum1+=num[i][j]; //主对角线元素累加
				   if(i+j==n-1)  
	                    sum2=sum2+num[i][j]; // 副对角线元素累加
			}
	}
	System.out.println("主对角线之和"+sum1);
	System.out.println("副对角线之和"+sum2);
	}
}