首页 > 学院 > 开发设计 > 正文

快速选择问题 【快排】

2019-11-08 01:07:52
字体:
来源:转载
供稿:网友

题目:快速选择问题

题意:输入n个整数和一个正整数k(1≤k≤n),输出这些整数从小到大排序后的第k个(例如,k=1就是最小值)。n≤10^7。 

思路:首先想到的是先排序然后输出对应下标即可,但是范围是10^7, 快排的速度是nlogn,如果限制速度的话就超时了!

我们可以在快排的过程中寻找k的范围,快排的是否要划分左右俩部分,这样我们每回划分时看k在左边还是右边,这样我们每次只需递归一半就可以了,只需排序k的范围内的数,其余的没有必要!

参考:紫书-快速排序-P227

代码:

#include <iostream>#include <cstdio>#include <algorithm>#include <cstring>using namespace std;const int maxn = 100;int n,k;int QKPass(int r[],int left,int right){//一次快排,即划分左右部分    int x = r[left];//基数    while(left < right){        while(left < right && r[right] >= x) right--;//从右端开始寻找一个比基数小的数        if(left < right) {r[left] = r[right]; left++;}//将小的数交换到左边        while(left < right && r[left] < x) left++;//从左端寻找比基数大的数        if(left < right) {r[right] = r[left]; right--;}//将大的数交换到右边    }    r[left] = x;//将基数放到中间部分    return left;//返回基数位置}void QKsortK(int r[],int left,int right){    if(left < right){        int pos = QKPass(r,left,right);//基数位置        if(k-1 <= pos - left +1) QKsortK(r,left,pos-1);//当k在左半部分时        else QKsortK(r,pos+1,right);//当k在右半部分时    }}int main(){    int a[maxn];    while(~scanf("%d%d",&n,&k)){        for(int i=0;i<n;i++) scanf("%d",&a[i]);        QKsortK(a,0,n-1);        PRintf("%d/n",a[k-1]);    }    return 0;}


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表