快速排序是一種基于分治的排序算法,其基本思想是選擇一個元素作為基準,將小于基準的元素放到基準的左邊,大于基準的元素放到基準的右邊,然后對左右兩個子數組分別進行快速排序。以下是用Python實現快速排序的代碼:
def quick_sort(arr):
if len(arr) <= 1:
return arr
else:
pivot = arr[0]
less = [x for x in arr[1:] if x <= pivot]
greater = [x for x in arr[1:] if x > pivot]
return quick_sort(less) + [pivot] + quick_sort(greater)
使用示例:
arr = [3, 1, 5, 2, 4]
sorted_arr = quick_sort(arr)
print(sorted_arr) # 輸出 [1, 2, 3, 4, 5]
在該實現中,我們選擇數組的第一個元素作為基準,并使用列表推導式將小于基準的元素放到less
列表中,大于基準的元素放到greater
列表中。然后,我們遞歸地對less
和greater
進行快速排序,并將結果合并后返回。
需要注意的是,這個實現中每次選擇第一個元素作為基準,可能會導致在某些特定情況下(比如數組已經是有序的)快速排序的效率下降。為了解決這個問題,可以選擇隨機的基準元素,或者進行優化,比如三數取中法、取隨機數等。