从一维数组创建排序的二维数组

def create_sorted_2d_array(arr, num_rows, num_cols):
    # Step 1: Sort the 1D array
    sorted_arr = sorted(arr)

    # Step 2: Determine the number of rows and columns
    num_elements = len(sorted_arr)
    if num_cols == 0:
        num_cols = num_elements // num_rows

    # Step 3: Create an empty 2D array
    result = [[0] * num_cols for _ in range(num_rows)]

    # Step 4: Fill the 2D array with sorted elements
    for i in range(num_elements):
        row = i // num_cols
        col = i % num_cols
        result[row][col] = sorted_arr[i]

    # Step 5: Return the filled 2D array
    return result

# Example usage
arr = [9, 2, 5, 1, 7, 4, 8, 3, 6]
num_rows = 3
num_cols = 3
result = create_sorted_2d_array(arr, num_rows, num_cols)
print(result)

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]