1. 程式人生 > >13_資料結構與演算法_快速排序_Python實現

13_資料結構與演算法_快速排序_Python實現

#Created By: Chen Da

"""
快速排序的思想:
    選擇基準值pivot將陣列分成兩個子陣列,小於基準值的元素和大於基準值的元素;這個過程稱為partition
    對兩個子陣列進行快速排序;
    合併結果。
"""

#一個簡單粗暴的遞迴快排
def quik_sort(array):
    if len(array) < 2:
        return array
    else:
        pivot_index = 0     #這裡簡單的將首個元素作為基準值
        pivot = array[pivot_index]
        less_part = [i for i in array[pivot_index+1:] if i <= pivot]
        great_part = [i for i in array[pivot_index+1:] if i > pivot]
        return quik_sort(less_part) + [pivot] + quik_sort(great_part)

def test_quik_sort():
    import random
    lyst = list(range(10))
    random.shuffle(lyst)
    assert quik_sort(lyst) == sorted(lyst)

if __name__ == "__main__":
    test_quik_sort()

上面的快速排序的版本過於簡單粗暴,存在一些問題,如less_part和great_part需要額外的陣列佔據空間進行儲存,並且兩者進行了兩次遍歷。