用Python只要3行代码实现快速排序？

def quick_sort(arr: list):
    """stable quicksort"""
    if len(arr) <= 1:
        return arr
    return quick_sort([i for i in arr[1:] if i <= arr[0]]) \
           + [arr[0]] \
           + quick_sort([i for i in arr[1:] if i > arr[0]])

其实用 lambda 表达式只需要一行代码，但PEP8不推荐这种编码风格（推荐使用 "def" 来定义函数），函数式一般在定义的函数内部使用。

 

上面的代码虽然简洁，但需要额外开辟空间存储大于/小于【主元】 pivot 的元素（这里pivot选择索引0的元素），再用分治策略分别递归对这两个数组排序。

下面代码是不需要重新开辟空间的方式，即在原始数组上进行排序操作。

def quick_sort_2(arr: list):
    """unstable quicksort"""
    _quick_sort(arr, 0, len(arr) - 1)


def _quick_sort(arr, low, high):
    pivot = arr[low]
    i = low
    j = high
    while i < j:
        while i < high and arr[i] <= pivot:
            i += 1
        while j > low and arr[j] > pivot:
            j -= 1
        if i < j:
            arr[i], arr[j] = arr[j], arr[i]
            continue
        arr[low], arr[j] = arr[j], arr[low]
        _quick_sort(arr, low, j - 1)
        _quick_sort(arr, j + 1, high)