Python实现归并排序算法

def merge_sort(arr):
    # 如果数组中只有一个元素，直接返回
    if len(arr) <= 1:
        return arr

    # 分成两个部分
    mid = len(arr) // 2
    left = arr[:mid]
    right = arr[mid:]

    # 分别对左右两个部分进行排序
    left = merge_sort(left)
    right = merge_sort(right)

    # 合并左右两个部分
    return merge(left, right)


def merge(left, right):
    # 初始化结果数组和左右两个子数组的起始下标
    result = []
    i = j = 0

    # 循环合并左右两个子数组
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            result.append(left[i])
            i += 1
        else:
            result.append(right[j])
            j += 1

    # 将剩余的元素加入结果数组
    result += left[i:]
    result += right[j:]

    # 返回结果数组
    return result

arr = [3, 5, 1, 4, 2, 6]
sorted_arr = merge_sort(arr)
print(sorted_arr)  # 输出 [1, 2, 3, 4, 5, 6]