Преобразование двумерного массива в одномерный Java

package prac1;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;
 
 
 
 
public class Prac1 {
 
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        List list = new ArrayList();
        
        int i, j = 0, rows, columns, N, k;
        int[][] array = null;
        
        System.out.print("Введите кол-во строк :\n> ");
        rows = sc.nextInt();
        
        System.out.print("Введите кол-во столбцов:\n> ");
        columns = sc.nextInt();
            
        
        System.out.print("Введите максимум для генератора чисел:\n> ");
        N = sc.nextInt();
        
        array = new int[rows][columns];
        int[][] array2 = new int[rows][rows*columns];
        java.util.Random random = new java.util.Random();
 
        for (i = 0; i < rows; ++i)
            for (j = 0; j < columns; ++j)
                array[i][j] = random.nextInt(N) + 1;
        System.out.println("Матрица:");
        
        for (i = 0; i < rows; ++i, System.out.println())
            for (j = 0; j < columns; ++j)
                System.out.print(array[i][j] + " ");
        
        /*for (i = 0; i < rows; ++i, System.out.println())
            for (j = 0; j < columns; ++j)
                list.add(array[i][j]);
                System.out.print("Список");
                System.out.print(list+"\n");*/
        
        int[][]copiedarray = Arrays.copyOf(array, array.length);
        
        for (j = 0; j < columns*rows; ++j)
            System.out.print(copiedarray[i][j] + " ");
        
}
}

Scanner sc = new Scanner(System.in);
 
        int rows, columns, index = 0;
 
        System.out.print("Введите кол-во строк :\n> ");
        rows = sc.nextInt();
 
        System.out.print("Введите кол-во столбцов:\n> ");
        columns = sc.nextInt();
 
        int[][] array = new int[rows][columns];
 
        System.out.print("Введите максимум для генератора чисел:\n> ");
        int N = sc.nextInt();
 
        int[] array2 = new int[rows * columns];
 
        System.out.println("Матрица:");
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < columns; ++j) {
                array[i][j] = ThreadLocalRandom.current().nextInt(N + 1);
                System.out.print(array[i][j] + " ");
                array2[index++] = array[i][j];
            }
            System.out.println();
        }
        System.out.println("\nArray2: " + Arrays.toString(array2));

import java.util.Arrays;
import java.util.concurrent.ThreadLocalRandom;
 
public class Helper {
    // генерит не кводратный 2-х мекрный массив, случайной длинны строки (от 1 до 6), заполненной случайными числам
   // параметр задает количество строк
    private static int[][] getNonSquarMatrix(int row) {
        int[][] ret = new int[row][];
        for (int i = 0; i < row; i++) {
            int col = ThreadLocalRandom.current().nextInt(1, 6);
            ret[i] = new int[col];
            for (int j = 0; j < col; j++) {
                ret[i][j] = ThreadLocalRandom.current().nextInt(1, 10);
            }
        }
        return ret;
    }
 
    // из названия понятно, что делает метод
    private static void print2DimArray(int[][] arr) {
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[i].length; j++) {
                System.out.print(j + 1 < arr[i].length ? "" + arr[i][j] + ", " : "" + arr[i][j] + '\n');
            }
        }
        System.out.println("______________");
    }
 
    // из не квадрат. 2-х мерного массива генерит одномерный
    private static int[] getOneDimArrFromTwoDimArray(int [][] matrix) {
        int countElem = 0;
        for (int [] tmpArr : matrix)
            countElem += tmpArr.length;
        int [] ret = new int[countElem];
        int indRet = 0;
        for (int [] tmpArr : matrix)
            for (int elemTmpArr : tmpArr)
                ret[indRet++] = elemTmpArr;
        return ret;
    }
 
    public static void main(String[] args) {
        int [][] twoNonSquDimArr = getNonSquarMatrix(4);
        print2DimArray(twoNonSquDimArr);
        int [] oneDimArrFromTwo = getOneDimArrFromTwoDimArray(twoNonSquDimArr);
        System.out.println(Arrays.toString(oneDimArrFromTwo));
    }
}