给定若干整数,请设计一个高效的算法,确定第k小的数。
输入格式:
测试数据有多组,处理到文件尾。每组测试数据的第1行输入2个整数n,k(1≤k≤n≤1000000)。第2行输入n个整数,每个数据的取值范围在0到1000000之间。
输出格式:
对于每组测试,输出第k小的数。
输入样例:
5 3
1 2 2 2 1
9 3
1 2 3 4 5 6 9 8 7
输出样例:
2
3
提示:
如果提交后超时,请注意需要设计的是高效的算法!如果你初学《数据结构》,暂时设计不出来,请学完相关知识再回头来做。
也可以使用STL之sort、nth_element函数求解。
另外,由于输入数据特别多,为减少输入数据的耗时,建议使用以下的输入外挂函数:
void scan_d(int &num)
{
char in;
in=getchar();
while(in<'0'||in>'9') in=getchar();
num=in-'0';
while(in=getchar(),in>='0'&&in<='9'){num*=10,num+=in-'0';}
}
调用方法如下:
int n;
scan_d(n);
或者使用scanf()函数进行输入;
或者使用以下语句提高cin、cout的效率:
ios::sync_with_stdio(false);
#include<bits/stdc++.h>
using namespace std;
void quicksort(int *arr,int l,int r){
if(l>=r)return;
int i=l,j=r;
int pri=arr[l];
while(i<j){
while(i<j&&arr[j]>=pri)j--;
if(i<j)arr[i++]=arr[j];
while(i<j&&arr[i]<pri)i++;
if(i<j)arr[j--]=arr[i];
}
arr[i]=pri;
quicksort(arr,l,i-1);
quicksort(arr,i+1,r);
}
int main(){
int n,k;
while(cin>>n>>k){
int *arr=(int *)malloc(n * sizeof(int));
for(int i=0;i<n;i++){
int x;
scanf("%d",&x);
arr[i]=x;
}
quicksort(arr,0,n-1);
printf("%d\n",arr[k-1]);
free(arr);
}
return 0;
}