POJ 2104 K-th Number（划分树）

题意：
求区间第k小

题解：
刚看划分树，感觉划分树给我的印象就是仅仅用来求区间第k大/小…

划分树模板题，稍微写了些注释在代码中，没什么好说的

---

AC代码：

#pragma GCC optimize(2)
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
using namespace __gnu_cxx;
typedef long long LL;
const int MAXN = 1e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int a[MAXN],sorted[MAXN],num[20][MAXN],val[20][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++){
        num[dep][i]=(i==l ? 0:num[dep][i-1]);
        if(val[dep][i]<sorted[mid] || (val[dep][i]==sorted[mid]&&same>0)){
            val[dep+1][ln++]=val[dep][i]; num[dep][i]++;
            if(val[dep][i]==sorted[mid]) same--;
        }else val[dep+1][rn++]=val[dep][i];
    }
    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(l==r) return val[dep][l];
    int ll=(ql==l ? 0:num[dep][ql-1]);
    int tot=num[dep][qr]-ll,mid=(l+r)>>1;
    if(tot>=k) return query(dep+1,l,mid,l+ll,l+num[dep][qr]-1,k);
    else{
        //ql-l表示ql前面有多少个数，减去ll表示这些数中取右子树的有多少个
        int lr=mid+1+(ql-l-ll);//lr表示去右边的开头位置
        //qr-ql+1表示区间总个数，减去去左边的，剩下的是去右边的
        return query(dep+1,mid+1,r,lr,lr+qr-ql+1-tot-1,k-tot);
    }
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    int n,m,l,r,k;
    while(~scanf("%d%d",&n,&m)){
        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);
        while(m--){
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",query(0,1,n,l,r,k));
        }
    }
    return 0;
}