主对角元素之和

 题目：从键盘输入一个整数n，生成n*n的方阵，输出主对角元素之和。

public class ZhuYuan{
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        System.out.println("请输入一个整数");
        int n = in.nextInt();
        int[][] num = new int[n][n];

        //生成矩阵
        int temp=1;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                num[i][j]=temp;
                temp++;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                System.out.print(num[i][j]+" ");
            }
            System.out.println();
        }
        int  a=0;
        int b=0;
        for(int i=0;i<num.length;i++){
            a+=num[i][i];
        }
        System.out.println("主对角线元素和为"+a);
    }

}