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

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

public class juzhen {

 public static void main(String[] args) {
  // TODO 自动生成的方法存根
  int sum=0;
  Scanner reader=new Scanner(System.in);
  System.out.println("请输入方阵的阶数");
  int n=reader.nextInt();
  int[][] a=new int[n][n];
  for(int i=0;i<n;i++)
  for(int j=0;j<n;j++)
  {
  System.out.println("输入第"+i+"行第"+j+"列元素的值");
  a[i][j]=reader.nextInt();
  }
  System.out.println("输入的"+n+"阶方阵为");
  for(int i=0;i<n;i++)
  { for(int j=0;j<n;j++)
  {
  System.out.print(a[i][j]+" ");
  }
  System.out.println();
  }
  for(int i=0;i<n;i++)
  {
  sum=sum+a[i][i];
  }

  System.out.print("对角线的和为："+sum);

  }

}