寻找第k小的数

给定若干整数,请设计一个高效的算法,确定第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;
}

  • 6
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值