【打卡】主对角线元素之和

描述
求一个$n\ast n$的整型矩阵 nums 的主对角线元素之和。请通过 System.out.println 语句输出矩阵 nums 的主对角线元素之和。
样例
评测机会将整个项目的代码编译为一个可执行的 Main 程序，并按照这样的方式执行你的代码 Main，你的代码需要从标准输入流（控制台）中读入数据 n 和 nums（共 n + 1 行，矩阵按行顺序输入，每行 n 个整数，以空格分隔），计算出结果后并打印到标准输出流（控制台）中。

样例 1：

当 n = 2，nums[0][0] = 1，nums[0][1] = 2，nums[1][0] = 3，nums[1][1] = 4 时，应该打印

5

样例 2：

当 n = 1，nums[0][0] = 9 时，应该打印

9

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        // write your code here
        // read data from console

        // output the answer to the console according to the
        // requirements of the question
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();

        int nums[][] = new int[n][n];
        for(int i=0; i < n; i++){
            for(int j=0; j < n; j++){
                nums[i][j] = sc.nextInt();
            }
        }

        int answer = 0;
        for(int i = 0; i < n; i++){
            answer += nums[i][i];
        }
        System.out.println(answer);
    }
}