三向切分的快速排序
对于含有大量重复元素的数组,使用三向切分快速排序堆快速排序进行改进。从左到右遍历数组一次,维护一个指针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的值(这样循环才会结束)。另外,除非和切分元素相等,其他元素都会被交换。