三路快排
阿新 • • 發佈:2018-12-26
三路快排也是用來解決序列中存在大量重複元素的問題,比雙路快排更高效
將序列分為三個部分,小於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]