Для заданной квадратной матрицы сформировать одномерный массив из ее диагональных элементов

import java.util.Scanner;
public class hi {
 public static void main (String [] args){
int n, mas[];
float s=0;
Scanner in = new Scanner(System.in);
System.out.print("Введите размерность массива: ");
n = in.nextInt();
mas = new int[n];
for (int i=0;i<mas.length; i++){
System.out.print("Введите "+i+"-й"+" элемент: ");
mas[i]=in.nextInt();
s+=mas[i];
}
s/=n;
System.out.println("Ср. арифм. массива: "+s);
}
}

    public static void main(String[] args) {
        int[][] arr = new int[][] {
                {1, 2, 3, 4},
                {5, 6, 7, 8},
                {9,10,11,12},
                {13,14,15,16}
        };
 
        for (int i = 0; i < arr.length; i++) {
            if (arr.length != arr[i].length) {
                throw new IllegalArgumentException("The matrix is not square!");
            }
        }
 
        int[] result = new int[arr.length * 2];
 
        for (int i = 0, j = arr.length - 1; i < arr.length; i++, j--) {
            result[i] = arr[i][i];
            result[i + arr.length] = arr[i][j];
        }
 
        System.out.println(Arrays.toString(result));    // [1, 6, 11, 16, 4, 7, 10, 13]
    }

    int[][] arr = new int[][] {
            {1, 2, 3, 4, 5},
            {6, 7, 8, 9,10},
            {11,12,13,14,15},
            {16,17,18,19,20},
            {21,22,23,24,25}
    };