目录
- 一、快速排序
- 二、Python算法实现
一、快速排序
快速排序的概念相信大家都能理解,下面这个算法是基于同样想法的另一种算法,其中利用到了分区。如果实施正确,这是一种非常有效的算法,在预期的O(n.log n) 时间内运行,乘法常数非常低,大约为2
需要注意的是,该算法的标准版本仅在唯一数据上是线性的。如果元素出现多次,性能会下降。它是一种3向快速排序,它将数据分成三个分区,较低、较高和类似于pivot区。另一个注意的地方就是枢轴的均匀随机化,这是证明算法预期在线性时间运行的重要部分
二、Python算法实现
我们首先来测试在排序时不使用这个排序算法的时间是多少呢?
import numpy as npdef swap(data, i, j):data[i], data[j] = data[j], data[i]
def qsort3(data, left, right):if left >= right:return# 选择中心点i = np.random.randint(left, right + 1)swap(data, left, i)pivot = data[left]# i表示左分区元素后的序号# j表示右分区元素后的序号# k表示当前元素的序号i, j, k = left, right, left + 1# 分割元素[left] + [pivot] + [right]while k <= j:if data[k] < pivot:swap(data, i, k)i += 1elif data[k] > pivot:swap(data, j, k)j -= 1k -= 1k += 1# 递归qsort3(data, left, i - 1)qsort3(data, j + 1, right)
def qsort(data):qsort3(data, 0, len(data) - 1)data = np.random.randint(0, 10, 100)
print(data)
看一下测试结果和运行时间
可以看到,此时数据是随机的,完成时间是1.7s
接下来我们把print(data)注释掉,先用算法排序之后再print
import numpy as npdef swap(data, i, j):data[i], data[j] = data[j], data[i]
def qsort3(data, left, right):if left >= right:returni = np.random.randint(left, right + 1)swap(data, left, i)pivot = data[left]i, j, k = left, right, left + 1while k <= j:if data[k] < pivot:swap(data, i, k)i += 1elif data[k] > pivot:swap(data, j, k)j -= 1k -= 1k += 1qsort3(data, left, i - 1)qsort3(data, j + 1, right)
def qsort(data):qsort3(data, 0, len(data) - 1)data = np.random.randint(0, 10, 100)
# print(data)qsort(data)
print(data)
测试结果
可以看到,在经过算法排序之后,再print(data),不仅正确排序,时间花费仅为0.5s,运行时间效率提高了