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需要额外的数组占据空间进行存储，并且两者进行了两次遍历。