算法导论程序14-快速排序的随机化版本（Python）

与始终采用A[r]作为主元的方法不同，随机抽样是从子数组A[p...r]中随机选择一个元素作为主元。

首先将A[r]与A[p...r]中随机选出的一个元素进行交换，通过对序列p,,,,r的随机抽样，我们可以保证主元元素x=A[r]是等概率地从子数组r-p+1个元素中选取的。因为主元元素是随机选

取的，我们期望在平均情况下，对输入数组的划分是比较均衡的。

def randomized_quicksort(A,p,r):    if p<r:        q=randomized_partition(A,p,r)        randomized_quicksort(A,p,q-1)        randomized_quicksort(A,q+1,r)import randomdef randomized_partition(A,p,r):    i = random.randint(p,r)    t=A[r]    A[r]=A[i]    A[i]=t    x=A[r]    i=p-1    for j in range(p,r):        if A[j]<x:            i+=1            t=A[i]            A[i]=A[j]            A[j]=t    A[r]=A[i+1]    A[i+1]=x    return i+1

运行结果：

>>> A=[2,8,7,1,3,5,6,4]>>> randomized_quicksort(A,0,7)>>> A[1, 2, 3, 4, 5, 6, 7, 8]

在输入元素互异的情况下，快速排序算法的期望运行时间为O(nlgn)