HDU-4417 Super Mario,划分树+二分！

Super Mario

这个题也做了一天，思路是很清晰，不过二分那里写残了，然后又是无限RE。。

题意：就是查询区间不大于k的数的个数。

思路：裸划分树+二分答案。将区间长度作为二分范围。这个是重点。没AC之前无限YY，AC之后却又觉得不过如此。

int a[N],tree[30][N],t_l[30][N];
void build(int l,int r,int id)
{
    if(l==r) return ;
    int mid=(l+r)/2,same=mid-l+1;
    for(int i=l;i<=r;i++) if(tree[id][i]<a[mid]) same--;
    int lp=l,rp=mid+1;
    for(int i=l;i<=r;i++)
    {
        if(tree[id][i]<a[mid]) tree[id+1][lp++]=tree[id][i];
        else if(tree[id][i]==a[mid]&&same) tree[id+1][lp++]=tree[id][i],same--;
        else tree[id+1][rp++]=tree[id][i];
        t_l[id][i]=t_l[id][l-1]+lp-l;
    }
    build(l,mid,id+1);
    build(mid+1,r,id+1);
}
//int num;C
int find(int l,int r,int k,int L,int R,int id)
{
    if(l==r) return tree[id][l];
    int mid=(L+R)/2;
    int cnt=t_l[id][r]-t_l[id][l-1];
    if(cnt>=k)
    {
        int newl=L+t_l[id][l-1]-t_l[id][L-1];
        int newr=newl+cnt-1;
        return find(newl,newr,k,L,mid,id+1);
    }
    else
    {
        int newr=r+t_l[id][R]-t_l[id][r];
        int newl=newr-(r-l-cnt);
        return find(newl,newr,k-cnt,mid+1,R,id+1);
    }
}
int main()
{
    int t,n,m;
    scanf("%d",&t);
    int t1=t;
    while(t--)
    {
        for(int i=0;i<30;i++) t_l[i][0]=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&tree[0][i]);
            a[i]=tree[0][i];
        }
        sort(a+1,a+1+n);
        build(1,n,0);
        int l,r,h;
        printf("Case %d:\n",t1-t);
        while(m--)
        {
            scanf("%d%d%d",&l,&r,&h);
            l++,r++;
            int ans=0,lp=1,rp=(r-l+1);//二分的范围
            while(lp<=rp)
            {
                int mid=(lp+rp)/2;
                int x=find(l,r,mid,1,n,0);
                if(x<=h)
                {
                    ans=mid;
                    lp=mid+1;
                }
                else rp=mid-1;
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}