如何将一个主数组的元素尽可能均衡地分配给另外两个数组？

def distribute_elements(main_array, array_length):
    total_length = len(main_array)
    array1_length = array_length
    array2_length = total_length - array1_length

    # 计算每个数组应该分配的元素数量
    array1_elements = total_length // 2
    array2_elements = total_length - array1_elements

    # 创建两个空数组
    array1 = []
    array2 = []

    # 遍历主数组，按照顺序分配元素给两个数组
    for i in range(total_length):
        if len(array1) < array1_elements:
            array1.append(main_array[i])
        else:
            array2.append(main_array[i])

    return array1, array2

# 测试
main_array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
array_length = 5
result1, result2 = distribute_elements(main_array, array_length)
print("Array 1:", result1)
print("Array 2:", result2)