双路快排

序列元素重复,是序列已经排好序的一种特殊情况,如果一个序列中的元素全部相同,也将出现最差情况。
如果序列中存在大量重复元素,在普通快排中,相等的元素会被全部放到分区点的一边,这样会大大增加快排的时间复杂度,双路快排就是用来解决这个问题的。
能够将序列均衡分开的分区点才是好的分区点。均匀分开意味着保持O(logn)的复杂度。

from random import shuffle, randrange

def quick_sort(lst, left, right):
    # 当只有一个元素的时候退出递归
    if left < right:
        p = partition(lst, left, right)
        # 左右分区分别递归
        quick_sort(lst, left, p)
        quick_sort(lst, p+1, right)

def partition(lst, left, right):
    rand = randrange(left, right)
    lst[left], lst[rand] = lst[rand], lst[left]   # 随机挑选出一个元素,与第一个元素交换,作为分区点
    pivot = lst[left]    # 以第一个元素为分区点
    leftpos = left + 1
    rightpos = right - 1
    while leftpos <= rightpos:
        while leftpos <= rightpos and lst[leftpos] <= pivot:
            leftpos += 1
        while leftpos <= rightpos and lst[rightpos] >= pivot:
            rightpos -= 1
        if leftpos > rightpos:
            break
        lst[leftpos], lst[rightpos] = lst[rightpos], lst[leftpos]
    # 将pivot放入分区点
    lst[leftpos-1], lst[left] = lst[left], lst[leftpos-1]
    # 返回分区点索引
    return leftpos-1


source = list(range(10))
shuffle(source)
print(source)

quick_sort(source, 0, len(source))
print(source)
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
posted @ 2018-12-26 01:07  KeithTt  阅读(502)  评论(0编辑  收藏  举报