算法导论程序13-快速排序的描述（Python）

快速排序：

采用分治思想：

1.分解：数组A[p....r]被划分成两个（可能为空）子数组A[p...q-1]和A[q+1....r]，使得A[p...q-1]中的每个元素都小于等于A[q]。而A[q]也小于等于A[q+1....r]中的每个元素。其中，计算下标q也是划分过程的一部分。

2.解决：通过递归调用快速排序，对子数组A[p...q-1]和A[q+1....r]进行排序。

3.合并：因为子数组都是原址排序的，所以不需要合并操作：数组A[p...r]已经有序。

def quicksort(A,p,r):
    if p<r:
        q=partition(A,p,r)
        quicksort(A,p,q-1)
        quicksort(A,q+1,r)



def partition(A,p,r):
    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]
>>> partition(A,0,7)
3
>>> A
[2, 1, 3, 4, 7, 5, 6, 8]
>>> quicksort(A,0,7)
>>> A
[1, 2, 3, 4, 5, 6, 7, 8]

重点就是：

partition方法中找到主元A[r]的正确位置。（为什么选择A[r]为主元?）

找到A[r]的正确位置：

不断比较A[r]之前的元素与A[r]的大小关系，如果比A[r]小，则，放到A[i]的位置上，i位置专门记录比A[r]小的元素。

如果元素比A[r]大，则不用动，相应的，i 的值就不变。

快速排序的平均运行时间：O(nlgn)