方阵对角线Java

题目：

老师让同学们排成了一个 n∗n 的方阵。好奇的你想知道这个方阵的对角线 上的同学的身高都是怎么样的。​ 现已知所有同学的身高，数值为整数，单位：厘米。要求输出所有对角线上（自左向右、自上 向下编号时行列编号相同）的同学的身高。

样例输入1

3
140 160 140
183 172 170
180 181 174

样例输出1

140
172
174

 代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int arr[][] = new int[n][n];
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr.length; j++) {
                arr[i][j] = scanner.nextInt();
            }
        }
        for (int i = 0; i < arr.length; i++) {
            System.out.println(arr[i][i]);

        }
    }
}