快速排序算法理解-python

快速排序定义

定义一个基准值（通常是第一个），将大于基准值的放在右边，小于基准值的放在右边。

举例

以2 1 9 6 7进行排序：
假设以2作为基准数，大的放右侧，小的放左侧
2 1 9 6 7
1 2 9 6 7 第一次比较
再以2后面的第一个数9作为基准数，因为9左侧的已经排好了
1 2 9 6 7
1 2 6 9 7第一次排序
1 2 6 7 9第二次排序

代码

def fast_sort(data):
    if len(data) >= 2:
        mid = data[0]  # 选取第一个数基准值
        left, right = [], []  # 定义基准值左右两侧的列表
        data.remove(mid)  # 从原始数组中移除基准值
        for num in data:
            if num >= mid:   #大于基准值的放右边
                right.append(num)
            else:
                left.append(num)   #小于基准值的放左边
        return fast_sort(left) + [mid] + fast_sort(right)
    else:
        return data

array = [2, 1, 9, 6, 7]
print(fast_sort(array))

运行结果：

参考：https://zhuanlan.zhihu.com/p/81434337