python实现归并排序(merge sort)

归并排序的思想就是先递归分解数组，再合并数组。将数组分解最小之后，然后合并两个有序数组，基本思路是比较两个数组的最前面的数，谁小就先取谁，取了后相应的指针就往后移一位。然后再比较，直至一个数组为空，最后把另一个数组的剩余部分复制过来即可。

下面的动态图非常贴切，记得点开从头看：

• 最优时间复杂度：O(nlogn)
• 最坏时间复杂度：O(nlogn)
• 稳定性：稳定

代码实现：

def merge_sort(alist):
    n = len(alist)  #获取当前传入的列表的长度
    if n <= 1:     # 如果列表长度为1，即已经拆分到最后一步了，就直接返回当前列表，不再往下继续执行了
        return alist
    mid = n // 2  # 取中间值，把当前输入的列表从中间拆分

    left_li = merge_sort(alist[:mid])   # 取左半部分
    right_li = merge_sort(alist[mid:])  # 取右半部分
    
    left_pointer = 0     # 设定左半部分的指针，从0开始
    right_pointer = 0    # 设定右半部分的指针，从0开始
    result = []          # 定义一个空列表result用于存储每次递归产生的排好序的列表

    while left_pointer < len(left_li) and right_pointer < len(right_li):  # 当各部分指针还没走到末尾时
        if left_li[left_pointer] <= right_li[right_pointer]:    # 把较小的值存入result并让相应的指针+1
            result.append(left_li[left_pointer])
            left_pointer += 1
        else:
            result.append(right_li[right_pointer])
            right_pointer += 1
    result += left_li[left_pointer:]   # 如果是奇数个元素，别忘了把最后单个的元素也添加到result里
    result += right_li[right_pointer:]
    return result   # 最终返回的是一个新的排好序的列表result，因此空间复杂度要多一倍

测试：

if __name__ == "__main__":
    li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
    print(li)
    li_sorted = merge_sort(li)
    print(li_sorted)

运行一下：

[54, 26, 93, 17, 77, 31, 44, 55, 20]
[17, 20, 26, 31, 44, 54, 55, 77, 93]