快速排序的四種python實現
- 2020 年 1 月 13 日
- 筆記
快速排序演算法,簡稱快排,是最實用的排序演算法,沒有之一,各大語言標準庫的排序函數也基本都是基於快排實現的。
本文用python語言介紹四種不同的快排實現。
1. 一行程式碼實現的簡潔版本
quick_sort = lambda array: array if len(array) <= 1 else quick_sort([item for item in array[1:] if item <= array[0]]) + [array[0]] + quick_sort([item for item in array[1:] if item > array[0]])
2. 網上常見的快排實現
def quick_sort(array, left, right): if left >= right: return low = left high = right key = array[low] while left < right: while left < right and array[right] > key: right -= 1 array[left] = array[right] while left < right and array[left] <= key: left += 1 array[right] = array[left] array[right] = key quick_sort(array, low, left - 1) quick_sort(array, left + 1, high)
由於快排是原地排序,因此不需要返回array。
array如果是個列表的話,可以通過len(array)求得長度,但是後邊遞歸調用的時候必須使用分片,而分片執行的原列表的複製操作,這樣就達不到原地排序的目的了,所以還是要傳上邊界和下邊界的。
3.《演算法導論》中的快排程式
def quick_sort(array, l, r): if l < r: q = partition(array, l, r) quick_sort(array, l, q - 1) quick_sort(array, q + 1, r) def partition(array, l, r): x = array[r] i = l - 1 for j in range(l, r): if array[j] <= x: i += 1 array[i], array[j] = array[j], array[i] array[i + 1], array[r] = array[r], array[i+1] return i + 1
這個版本跟上個版本的不同在於分片過程不同,只用了一層循環,並且一趟就完成分片,相比之下程式碼要簡潔的多了。
4. 用棧實現非遞歸的快排程式
先說兩句題外話,一般意義上的棧有兩層含義,一層是後進先出的數據結構棧,一層是指函數的記憶體棧,歸根結底,函數的記憶體棧的結構就是一個後進先出的棧。彙編程式碼中,調用一個函數的時候,修改的也是堆棧指針暫存器ESP,該暫存器保存的是函數局部棧的棧頂,另外一個暫存器EBP保存的是棧底。不知道與棧存儲空間相對的堆存儲空間,其組織結構是否也是一個完全二叉樹呢?
高級語言將遞歸轉換為迭代,用的也是棧,需要考慮兩個問題:
1)棧裡邊保存什麼?
2)迭代結束的條件是什麼?
棧裡邊保存的當然是需要迭代的函數參數,結束條件也是跟需要迭代的參數有關。對於快速排序來說,迭代的參數是數組的上邊界low和下邊界high,迭代結束的條件是low == high。
def quick_sort(array, l, r): if l >= r: return stack = [] stack.append(l) stack.append(r) while stack: low = stack.pop(0) high = stack.pop(0) if high - low <= 0: continue x = array[high] i = low - 1 for j in range(low, high): if array[j] <= x: i += 1 array[i], array[j] = array[j], array[i] array[i + 1], array[high] = array[high], array[i + 1] stack.extend([low, i, i + 2, high])
另外,當數組下標為-1時,C++、Java等語言中會報錯,但python中訪問的是最後一個元素,所以如果程式寫錯了,可能其他語言會報錯,但python會輸出一個錯誤的結果。