HDU 4417 Super Mario（划分树）

题意：
给出一个长度为n的序列，Q次询问l，r，H，询问在[l,r]区间中小于H的数有多少个？

---

题解：
用划分树求第k小的，然后二分答案即可

---

AC代码：

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
typedef long long LL;
typedef pair<int,int> pii;
const int MAXN = 1e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int num[20][MAXN],val[20][MAXN],a[MAXN],sorted[MAXN];
inline void build(int l,int r,int dep){
    if(l==r) return;
    int mid=(l+r)>>1,same=mid-l+1;
    for(int i=l;i<=r;i++) if(val[dep][i]<sorted[mid]) same--;
    int ln=l,rn=mid+1;
    for(int i=l;i<=r;i++){
        if(val[dep][i]<sorted[mid] || (val[dep][i]==sorted[mid] && same>0)){
            val[dep+1][ln++]=val[dep][i];
            if(val[dep][i]==sorted[mid]) same--;
        }else val[dep+1][rn++]=val[dep][i];
        num[dep][i]=num[dep][l-1]+ln-l;
    }
    build(l,mid,dep+1); build(mid+1,r,dep+1);
}
inline int query(int dep,int l,int r,int ql,int qr,int k){
    if(ql==qr) return val[dep][ql];
    int mid=(l+r)>>1,tot=num[dep][qr]-num[dep][ql-1];
    if(tot>=k){
        int ll=l+num[dep][ql-1]-num[dep][l-1];
        int lr=ll+tot-1;
        return query(dep+1,l,mid,ll,lr,k);
    }else{
        int lr=qr+num[dep][r]-num[dep][qr];
        int ll=lr-(qr-ql+1-tot)+1;
        return query(dep+1,mid+1,r,ll,lr,k-tot);
    }
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    int T,cas=0; scanf("%d",&T);
    while(T--){
        int n,Q; scanf("%d%d",&n,&Q);
        for(int i=1;i<=n;i++) scanf("%d",&val[0][i]),sorted[i]=val[0][i];
        sort(sorted+1,sorted+n+1);
        build(1,n,0);
        printf("Case %d:\n",++cas);
        while(Q--){
            int l,r,k; scanf("%d%d%d",&l,&r,&k); l++;r++;
            int ll=1,rr=r-l+1,ans=0;
            while(ll<=rr){
                int mid=(ll+rr)>>1;
                if(query(0,1,n,l,r,mid)<=k) ll=mid+1,ans=mid;
                else rr=mid-1;
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}