与始终采用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 random
def 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]
扫一扫
2979
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
