将2D Array的列更改为1D Array java

public class Main {
    public static void main(String[] args) {
        int[][] twoDArray = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};

        int[] oneDArray = convertTo1DArray(twoDArray);

        // 打印转换后的一维数组
        for (int num : oneDArray) {
            System.out.print(num + " ");
        }
    }

    public static int[] convertTo1DArray(int[][] twoDArray) {
        int rows = twoDArray.length;
        int cols = twoDArray[0].length;

        int[] oneDArray = new int[rows * cols];

        int index = 0;
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                oneDArray[index] = twoDArray[i][j];
                index++;
            }
        }

        return oneDArray;
    }
}