求区间第k大

int a[MX];
void insert(int a[],int l,int r)
{
    for(int i=l+1,j;i<=r;++i)
    {
        if(a[i]>a[i-1]) continue;
        int tmp=a[i];
        for(j=i;j>l&&a[j-1]>tmp;--j) a[j]=a[j-1];
        a[j]=tmp;
    }
}
int divide(int a[],int l, int r)//划分子问题
{
    int x = a[l];
    while (l < r)
    {
        while (l < r && a[r] >= x) r--;
        if (l < r) a[l++] = a[r];
        while (l < r && a[l] < x) l++;
        if (l < r) a[r--] = a[l];
    }
    a[l] = x;
    return l;
}
int Select(int s[],int l,int r,int k,int len)//返回s数组l~r的第k大数的下标
{
    //边界条件,s数组不足k个,直接排序返回
    if(r-l+1<=k)
    {
        insert(s,l,r);
        return l+k-1;
    }
    while(1)
    {
        //将s分为k个一组,共(r-l+1)/k组
        //每组排序,中位数放到集合M,这里直接将中位数交换到s数组前面,方便计算
        int n = r-l+1;
        for(int i=1;i<=n/r;++i)
        {
            insert(s,l+(i-1)*len,l+i*len-1);
            swap(s[l+i-1],a[l+(i-1)*len+r/2]);
        }
        //从集合M中选出中位数m*(递归)
        //以m*作为基准元素,划分子问题,这里直接将m*换到区间左端点,依然是为了方便计算
        int tag = Select(s,l,l+n/r-1,n/r/2+n/r%2,len);
        swap(s[l],s[tag]);
        tag = divide(s,l,r);
        if(tag-l+1==k) return tag;
        else if(tag-l+1>k) r=tag-1;
        else k-=(tag-l+1),l=tag+1;
    }
}
signed main()
{
    int n,k;cin>>n>>k;
    rpp(i,n) cin>>a[i];
    cout<<a[Select(a,1,n,k,5)]<<endl;;
    stop;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值