Fill the Matrix Patterns

package com.company;


import java.io.IOException;


import java.util.*;

public class Main {


    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        String[] input = scanner.nextLine().split(", ");
        int size = Integer.parseInt(input[0]);
        String pattern = input[1];

        int[][] matrix = new int[size][];

        if (pattern.equals("A")) {
            matrix = getMatrixA(size);
        } else {
            matrix = getMatrixB(size);
        }
        PrintMatrix(matrix);


    }

    private static void PrintMatrix(int[][] matrix) {
        for (int row = 0; row < matrix.length; row++) {
            for (int col = 0; col < matrix[row].length; col++) {
                System.out.print(matrix[row][col] + " ");
            }
            System.out.println();
        }
    }

    private static int[][] getMatrixA(int size) {
        int[][] matrix = new int[size][size];

        int value = 1;//започваме матрицата от едно по условие
        for (int col = 0; col < size; col++) {
            for (int row = 0; row < size; row++) {// ходим до size и в двата цикъла защото е квадратна матрица
                matrix[row][col] = value++;//увеличава се стойността с 1 по условие
            }
        }


        return matrix;
    }

    private static int[][] getMatrixB(int size) {
        int[][] matrix = new int[size][size];

        int value = 1;//започваме матрицата от едно по условие
        for (int col = 0; col < size; col++) {
            if (col % 2 == 0) {


                for (int row = 0; row < size; row++) {// ходим до size и в двата цикъла защото е квадратна матрица
                    matrix[row][col] = value++;//увеличава се стойността с 1 по условие
                }
            } else {
                for (int row = size - 1; row >= 0; row--) {
                    matrix[row][col] = value++;
                }
            }
        }


        return matrix;
    }

}