图示经典算法--三向切分的快速排序

三向切分的快速排序

对于含有大量重复元素的数组，使用三向切分快速排序堆快速排序进行改进。从左到右遍历数组一次，维护一个指针lt使得a[lo..lt-1]中的元素小于，一个指针gt使得a[gt+1..hi]中的元素都大于v，一个指针i使a[lt..i-1]中的元素都等于v，a[i..gt]中的元素都还未确定。

• a[i]小于v，将a[lt]和a[i]交换，将lt和i减一；
• a[i]大于v，将a[gt]和a[i]交换，将gt减一；
• a[i]等于v，将i将一。

class Quick3way():
    """三向切分的快速排序"""

    @classmethod
    def sort(cls, a):
        cls.sort_lh(a, 0, len(a) - 1)

    @classmethod
    def sort_lh(cls, a, lo, hi):
        if hi <= lo:
            return
        lt = lo
        i = lo + 1
        gt = hi
        v = a[lo]
        while i <= gt:
            if a[i] < v:
                a[lt], a[i] = a[i], a[lt]
                lt += 1
                i += 1
            elif a[i] > v:
                a[i], a[gt] = a[gt], a[i]
                gt -= 1
            else:
                i += 1
        # 结果a[lo..lt - 1] < v = a[lt..gt] < a[gt + 1..hi]
        cls.sort_lh(a, lo, lt - 1)
        cls.sort_lh(a, gt + 1, hi)

这些操作都会保证数组元素不变且缩小gt-i的值（这样循环才会结束）。另外，除非和切分元素相等，其他元素都会被交换。

示例代码