寻找无序数组中第k大的数——快排思想

找数组中第k大的数，避免o(n2)时间，考虑快排方法。

#include <iostream>
#include <cstdlib>
using namespace std;

int random_partion(int *p,int n)
{
	int idx=rand()%n; //随机取划分元素 
	swap(p[idx],p[n-1]); //将划分元素置于数组尾端 
	int i=-1; //标记最大的 小于标记元素的 元素位置 
	int j=0; //遍历数组 
	for(j=0;j<n;j++)
	{
		if(p[j]<p[n-1])
		{
			swap(p[++i],p[j]); //将小于划分元素的数据交换到左边 
		}
	}
	swap(p[++i],p[n-1]); //将划分元素放置到数组中间 
	return i;
}

int getMax(int *p,int n,int k)
{
	int mid;
	if(k<=0) return -1;
	if(n<k) return -1;
	mid=random_partion(p,n); 
	if(mid==n-k) return p[mid];  //如果划分元素位置恰好在n-k上，说明对应数组元素为第k大 
	else if(mid<n-k) getMax(p+mid+1,n-mid-1,k); //若划分元素位置小于n-k，则第k大数在右边数组中，且仍是第k大的数 
	else getMax(p,mid,k-(n-mid)); //若划分元素位置大于n-k，则第k大数在左边数组中，是第k-(n-mid)大的数 
}
int main()
{
	int n,k;
	while(cin>>n>>k) //数组长度 及 k 
	{
		int a[n+5];
		for(int i=0;i<n;i++)
		{
			cin>>a[i];
		}
		int ans;
		ans=getMax(a,n,k);
		cout<<ans<<endl;
	}
}