python 实现归并排序

时间:2022-12-22 23:15:05
import random
# 我也是想了很久才想明白,递归用的太少了。画下图自己模拟一下就行了。
# 先递归分成单个元素,然后排序。
# 排序的函数传入的值是两个连续的数组。
# 因为单个元素的时候就进行排序,所以返回的一left_half,right_half 一定是排好序的


def merge_sort(seq):
    if len(seq) <=1: # 只有一个元素时是递归出口
        return seq

    else:
        mid = int(len(seq)/2)
        left_half = merge_sort(seq[:mid])
        right_half = merge_sort(seq[mid:])

        # 合并两个有序数组
        new_seq = merge_sorted_list(left_half, right_half)
        return new_seq

'''
排序函数运用两个指针. 分别比较指针指向的大小,添加进新的数组。当两个数组不一样大的时候,因为之前已经是排好序的了,
直接按顺序添加进新的数组就完事了。
'''

def merge_sorted_list(sorted_a, sorted_b):
    length_a  = len(sorted_a)
    length_b = len(sorted_b)
    a = b = 0
    new_sorted_seq = list()
    while a < length_a and b < length_b:
        if sorted_a[a] < sorted_b[b]:
            new_sorted_seq.append(sorted_a[a])
            a += 1
        else:
            new_sorted_seq.append(sorted_b[b])
            b += 1

    while a<length_a:
        new_sorted_seq.append(sorted_a[a])
        a += 1

    while b<length_b:
        new_sorted_seq.append(sorted_b[b])
        b += 1


    return new_sorted_seq


l = [x for x in range(10000)]
random.shuffle(l)
print(l)
sorted_l = merge_sort(l)
print(sorted_l)