三路快排

三路快排也是用来解决序列中存在大量重复元素的问题,比双路快排更高效
将序列分为三个部分,小于pivot、等于pivot、大于pivot
等于pivot的部分不递归,这样在存在大量重复元素时,将大大缩小递归的数据规模

from random import shuffle, randint

def quick_sort(lst, left, right):
    if left < right:  # 当只有一个元素的时候退出递归
        rand = randint(left, right)
        lst[left], lst[rand] = lst[rand], lst[left]  # 随机挑选出一个元素,与第一个元素交换,作为分区点
        pivot = lst[left]  # 以第一个元素为分区点
        leftpos = left  # lst[left+1...lt] < v
        rightpos = right + 1    # lst[gt...right] > v
        i = left + 1    # lst[lt+1...i] == v
        while i < rightpos:
            if lst[i] < pivot:
                lst[leftpos + 1], lst[i] = lst[i], lst[leftpos + 1]
                leftpos += 1
                i += 1
            elif lst[i] > pivot:
                lst[rightpos - 1], lst[i] = lst[i], lst[rightpos - 1]
                rightpos -= 1
            else:
                i += 1
        lst[leftpos], lst[left] = lst[left], lst[leftpos]
        leftpos -= 1

        quick_sort(lst, left, leftpos)
        quick_sort(lst, rightpos, right)

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

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