小刘学python——快速排序

快速排序–python实现

快速排序的概念相信大家都很容易理解，下面这个算法是基于同样想法的另一种算法，其中利用到了分区。如果实施正确，这是一种非常有效的算法，在预期的O(nlog n) 时间内运行，乘法常数非常低，大约为2。

需要注意的是，该算法的标准版本仅在唯一数据上是线性的。如果元素出现多次，性能会下降。救生器是一种3向快速排序，它将数据分成三个分区，较低、较高和类似于pivot区。另一个注意的地方就是枢轴的均匀随机化，这是证明算法预期在线性时间运行的重要部分。
算法实现

import numpy as np
def swap(data, i, j):
    data[i], data[j] = data[j], data[i]
def qsort3(data, left, right):
    # sorted
    if left >= right:
        return
    # select pivot
    i = np.random.randint(left, right + 1)
    swap(data, left, i)
    pivot = data[left]
    # i ~ points behind left partition
    # j ~ points ahead of right partition
    # k ~ current element
    i, j, k = left, right, left + 1
    # split to [left] + [pivot] + [right]
    while k <= j:
        if data[k] < pivot:
            swap(data, i, k)
            i += 1
        elif data[k] > pivot:
            swap(data, j, k)
            j -= 1
            k -= 1
        k += 1
    # recursion
    qsort3(data, left, i - 1)
    qsort3(data, j + 1, right)
def qsort(data):
    qsort3(data, 0, len(data) - 1)
  

测试：

qsort(data)
print(data)