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;
}