QuickSort 和 QuickSelection中partition注记

来源：互联网发布：上古卷轴5原始捏脸数据编辑：程序博客网时间：2024/06/07 09:56

CTCI 5th中提到了quicksort partition中的另一种写法，感觉比算法导论上的更容易理解，更直观。

其中需要注意的是：

1) while(A[left] < pivot) left++; 该处不能使用==符号，否则所有元素相等时，没有办法进行partition，将会出现infinate looping

2) pivot的选取，可以选取[start, end]中任何一个，或者随机选取。

3) return left-1，是partition的左半部分的最有一个元素的Index。

Quick Partition代码如下：

#include <cstdlib>#include <cstdio>#include <iostream>#include <algorithm>using namespace std;int partition(int A[], int s, int e){int len = e-s+1;int pivot = A[rand()%(e-s+1)+s];int left = s, right = e;while(left <= right){while(A[left] < pivot) left++;while(A[right] > pivot) right--;if(left <= right){swap(A[left], A[right]);left++;right--;}}return left-1;}int select_kth(int A[], int s, int e, int k){int left_last = partition(A, s, e);int left_index = left_last - s;if(k == left_index)return A[left_last];else if(left_index > k)return select_kth(A, s, left_last, k);elsereturn select_kth(A, left_last+1, e, k-(left_index+1));}int main(){    int n = 5000;    int a[n];    int b[n];    for(int i=0; i<n; i++)    {        a[i] = rand()%10000;        b[i] = a[i];    }    cout << endl;    sort(b, b+n);    for(int i=0; i<n; i++)    {        if(select_kth(a,0,n-1,i) != b[i])        cout << "error " << endl;    }    return 0;}

Quick Sort代码如下：

0 0