HDU 2665 Kth number 划分树

题目大意:给定一个序列,求区间第k小

注意是第k小!!别被题目描述骗到了!!这题求的是第k小!!不是第k大!!!

这题和POJ2104一样,都是求区间第k小,不同的是这题的序列是有重复的

对于有重复的,我们必须先预处理出有多少个中位数能进入左区间,否则就会导致过多的中位数堆积在左区间导致该进入左区间的东西被硬塞进了右区间

其实我只是想说为何网上的处理重复都写的那么麻烦。。。像我这样精简点不好么?

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
int n,m,a[M],b[M],c[M];
int s[20][M];
void Build_Tree(int l,int r,int dpt)
{
    int i,mid=l+r>>1;
    if(l==r)
        return ;
    int l1=l,l2=mid+1,left=mid-l+1;
    for(i=l;i<=r;i++)
        left-=(a[i]<c[mid]);
    for(i=l;i<=r;i++)
    {
        if(a[i]>c[mid]||a[i]==c[mid]&&!left)
            b[l2++]=a[i],s[dpt][i]=(i==l?0:s[dpt][i-1]  );
        else
            b[l1++]=a[i],s[dpt][i]=(i==l?1:s[dpt][i-1]+1),left-=(a[i]==c[mid]);
    }
    memcpy( a+l , b+l , sizeof(a[0])*(r-l+1) );
    Build_Tree(l,mid,dpt+1);
    Build_Tree(mid+1,r,dpt+1);
}
int getans(int l,int r,int dpt,int x,int y,int k)
{
    int mid=l+r>>1;
    if(l==r)
        return c[mid];
    int l1=(x==l?0:s[dpt][x-1]),l2=s[dpt][y];
    if(k<=l2-l1)
        return getans(l,mid,dpt+1,l+l1,l+l2-1,k);
    else
        return getans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k-l2+l1);
}
int main()
{
    int T,i,x,y,k;
    for(cin>>T;T;T--)
    {
        cin>>n>>m;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]),c[i]=a[i];
        sort(c+1,c+n+1);
        Build_Tree(1,n,0);
        for(i=1;i<=m;i++)
            scanf("%d%d%d",&x,&y,&k),printf("%d\n", getans(1,n,0,x,y,k ) );
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值