POJ2104 K-th Number 不带修改的主席树 线段树

http://poj.org/problem?id=2104

给定一个序列，求区间第k小

通过构建可持久化的点，得到线段树左儿子和右儿子的前缀和（前缀是这个序列从左到右意义上的），然后是一个二分的get操作。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define LL long long
const LL maxn=100010;
using namespace std;
int n,m;
struct node{
    int lc,rc,sum;
}t[maxn*20];int tot;
struct nod{
    int x,id;
}a[maxn];
int rk[maxn]={},rt[maxn]={};
bool mcmp(nod aa,nod bb){return aa.x<bb.x;}
void insert(int &x,int l,int r,int num){
    t[++tot]=t[x]; ++t[tot].sum;x=tot;//cout<<t[x].sum<<endl;
    if(l==r)return;
    int mid=(r+l)/2;
    if(num<=mid)insert(t[x].lc,l,mid,num);
    else insert(t[x].rc,mid+1,r,num);
}
int getit(int x,int y,int k,int l,int r){
    if(l==r)return l;
    int z=t[t[y].lc].sum-t[t[x].lc].sum;
    int mid=(r+l)/2;
    if(k<=z)return getit(t[x].lc,t[y].lc,k,l,mid);
    else return getit(t[x].rc,t[y].rc,k-z,mid+1,r);
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        tot=0;
        for(int i=1;i<=n;i++){scanf("%d",&a[i].x);a[i].id=i;}
        sort(a+1,a+1+n,mcmp);
        for(int i=1;i<=n;i++) rk[a[i].id]=i; 
        for(int i=1;i<=n;i++){rt[i]=rt[i-1]; insert(rt[i],1,n,rk[i]);}
        int x,y,k;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&x,&y,&k);
            printf("%d\n",a[getit(rt[x-1],rt[y],k,1,n)].x);
        }
    }
    return 0;
}