Codeup100000587问题 A: 求第k大数

Codeup100000587问题 A: 求第k大数

题目描述：

给定一个长度为n（1≤n≤1,000,000）的无序正整数序列，以及另一个数k（1≤k≤1,000,000）（关于第k大的数：例如序列{1,2,3,4,5,6}中第3大的数是4。）

输入：

第一行两个正整数m,n。

第二行为n个正整数。

输出：

第k大的数。

样例输入：

6 3
1 2 3 4 5 6

样例输出：

4

实现代码：

#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <cstring>
#include <algorithm>
#include <math.h>
using namespace std;

int randPartition(int A[],int left,int right){
	int p=round(1.0*rand()/RAND_MAX*(right-left)+left);
	swap(A[left],A[p]);
	int temp=A[left];
	while(left<right){
		while(left<right&&A[right]<temp){
			right--;
		}
		A[left]=A[right];
		while(left<right&&A[left]>temp){
			left++;
		}
		A[right]=A[left];
	}
	A[left]=temp;
	return left;
}

int randSelect(int A[],int left,int right,int K){
	if(left==right){
		return A[left];
	}
	int p=randPartition(A,left,right);
	int M=p-left+1;
	if(M==K){
		return A[p];
	}else if(K<M){
		return randSelect(A,left,p-1,K);
	}else{
		return randSelect(A,p+1,right,K-M);
	}
	
}

int main()
{
	int m,n;
	int num[100010];
	srand((unsigned)time(NULL));
	while(scanf("%d%d",&m,&n)!=EOF){
		for(int i=0;i<m;i++){
			scanf("%d",&num[i]);
		}
		int ans=randSelect(num,0,m-1,n);
		printf("%d\n",ans);
	}
	return 0;
}