Вывести все числа, которые в обеих таблицах записаны в одних и тех же клетках

import java.util.Scanner;
 
public class Test {
    public static void main(String args[]) {
        int n = 0;
        try (Scanner in = new Scanner(System.in)) {
            do {
                System.out.print("n = ");
                n = in.nextInt();
            } while (n < 1 || n > 100);
        }
 
        int[][] a = new int[n][n];
        int[][] b = new int[n][n];
 
        int num = 1;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                a[i][j] = num;
                b[j][i] = num;
                num++;
            }
        }
 
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (a[i][j] == b[i][j]) {
                    System.out.print(String.valueOf(a[i][j]).concat(" "));
                }
            }
        }
    }
}

import java.util.Scanner;
 
public class Test {
    public static void main(String args[]) {
        int n = 0;
        try (Scanner in = new Scanner(System.in)) {
            do {
                System.out.print("n = ");
                n = in.nextInt();
            } while (n < 1 || n > 100);
        }
 
        for (int i = 1; i <= n * n; i++) {
            if (i / n == i % n - 1 || i == n * n) {
                System.out.print(String.valueOf(i).concat(" "));
            }
        }
    }
}

IntStream.range(0, n).map(i -> 1 + i * (n + 1)).forEach(i -> System.out.print(i + " "));

public class TestClass {
 
    static int n;
    
    public static void main(String[] args) throws IOException {
        try (Scanner in = new Scanner(System.in)) {
            do {
                System.out.print("n = ");
                n = in.nextInt();
            } while (n < 1 || n > 100);
        }
        // одной строкой
        IntStream.range(0, n).map(i -> 1 + i * (n + 1)).forEach(i -> System.out.print(i + " "));
        System.out.println("");
        // или циклом
        for (int i = 0; i < n; i++) {
            System.out.println(1 + i * (n + 1));
        }
        System.out.println("");
        
    }    
 
}