hihocoder1133(二分求第k小)

来源：互联网发布：淘宝装修怎么弄全屏编辑：程序博客网时间：2024/06/06 03:25

链接：点击打开链接

题意：求无序数组的第k小

代码：

#include <queue>#include <vector>#include <stdio.h>#include <stdlib.h>#include <string.h>#include <iostream>#include <algorithm>using namespace std;int n,k,ans,a[1000005];int change(int ll,int rr){    int tt=a[ll];    while(ll<rr){        while(ll<rr&&a[rr]>tt)        rr--;        a[ll]=a[rr];        while(ll<rr&&a[ll]<tt)        ll++;        a[rr]=a[ll];    }    a[ll]=tt;    return ll;}void cal(int ll,int rr){    int op=change(ll,rr);    if(op==k){        ans=a[op];        return;    }    else if(op>k)    cal(ll,op-1);    else    cal(op+1,rr);}int main(){                                     //主要思想还是基于快排，就是在排序的时候    int i;                                      //根据左右区间的大小从而只在一个区间继续排序    while(scanf("%d%d",&n,&k)!=EOF){            //这样做相比直接排序求第k小的复杂度的常数要小        for(i=1;i<=n;i++)                       //期望复杂度为O(n+n/2+n/4+...+1)=O(2N)        scanf("%d",&a[i]);                      //但是既然是基于快排的思想，我们依然可以构造        if(k<=0||k>n){                          //一个递增或递减的数列叉掉这个算法            puts("-1");                         //所以我们依然可以先随机化一下再进行操作            continue;        }        cal(1,n);        printf("%d\n",ans);    }    return 0;}

阅读全文

0 0