快速选择算法

快速选择算法可快速选出数组中第N大的数，采用的是分治的思想，效率为O(n)，poj 2388可做例题

// 快速选择算法
#include <cstdio>
 
const int maxn = 10000 + 5;
int a[maxn],n,k;
int Quick_select(int b,int e){
	int i=b,j=e,mid=a[(i+j)>>1];
	while(i<=j){
		while(a[i]<mid) i++;
		while(a[j]>mid) j--;
		if(i<=j){
			int t = a[i];
			a[i] = a[j];
			a[j] = t;
			i++;
			j--;
		}
	}
	if(j>b && k<=j) return Quick_select(b,j);
	if(i<e && k>=i) return Quick_select(i,e);
	return a[k];  // 已经排好序 直接返回 
} 

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	printf("第 %d 大的数是 %d\n",k,Quick_select(1,n));
	return 0;
}