查找第K大的元素

给定一个数组，找出这个数组的最小元素和最大元素我们只需要O(n)，但是如果要找第k大的元素，我们可以用暴力的方式搜索，进行k轮查找，第一轮查找第1大，第二轮查找第2大，第三轮查找第3大，...，第k轮查找第k大的元素，(n-1)+(n-2)+...+(n-k) = k*n-(k+1)/2，这种方式的效率不高。我们用类似于快排的方式将数组进行划分成两个数组，如果左边元素个数大于k,则在数组左边查找，如果左边元素小于k，则在右边查找。平均情况下，T(n) = T(n/2)  + (n), 最终平均复杂度位O(n)。

typedef int Item;
#define swap(a,b) {Item t = a; a = b; b = t; }

int partition(Item a[], int p, int r) {
	int m = p;
	for(int i = p+1; i <= r; i++) {
		if(a[i] < a[p]) {
			++m;
			swap(a[m],a[i]);
		}
	}
	swap(a[p],a[m]);

	return m;
}

int randomizedPartition(Item a[], int p, int r){
	int randNum = rand()%(r-p+1) + p; 
	swap(a[p],a[randNum]);
	return partition(a,p,r);
}

Item randomizedSelect(Item a[],int p, int r, int k)
{
	if(k > r-p+1) throw runtime_error("k is out of range!");
	if(p == r) return a[p];
	int q = randomizedPartition(a, p, r);
	if(k == q-p+1) {
		return a[q];
	}
	else if(k < q-p+1)
		return randomizedSelect(a, p,q-1,k);
	else 
		return randomizedSelect(a, q+1,r,k - (q-p+1));
}