[POJ2104/HDU2665]Kth Number-主席树-可持久化线段树

Problem Kth Number

Solution

裸的主席树，模板题。但是求k大的时候需要非常注意，很多容易写错的地方。卡了好久。写到最后还给我来个卡空间。 
具体做法参见主席树论文《可持久化数据结构研究》。

AC Code

#include "cstdio"
#include "iostream"
#include "cstring"
#include "algorithm"
using namespace std;
int T,a[],root[],b[],l,r,k,tot;
struct discre{
    int num,sum;
}dctz[];
struct tree{
    int lc,rc,sum;
}tr[*];
bool cmp(discre a,discre b){
    return a.sum<b.sum;
}
void build(int now,int l,int r){
    int mid=(l+r)/;
    tr[now].sum=;
    if(l==r){
        return;
    }
    tr[now].lc=++tot;
    build(tot,l,mid);
    tr[now].rc=++tot;
    build(tot,mid+,r);
}
void insert(int last,int sum,int l,int r){
    tr[++tot]=tr[last];
    tr[tot].sum++;
    int x=tot;
    if(l==r){
        return;
    }
    int mid=(l+r)/;
    if(sum>mid){
        tr[tot].rc=tot+;
        insert(tr[last].rc,sum,mid+,r);
    }else{
        tr[tot].lc=tot+;
        insert(tr[last].lc,sum,l,mid);
    }
}
int search(int k,int l,int r,int lx,int rx){
    if(lx==rx){
        return lx;
    }
    int mid=(lx+rx)/;
    int x=tr[tr[r].lc].sum-tr[tr[l].lc].sum;
    if((x>=k))return search(k,tr[l].lc,tr[r].lc,lx,mid);
    else return search(k-x,tr[l].rc,tr[r].rc,mid+,rx);
}
int main(){
    freopen("chairmantree.in","r",stdin);
    scanf("%d",&T);
    while(T--){
        memset(tr,sizeof(tr),);
        memset(a,sizeof(a),);
        memset(dctz,sizeof(dctz),);
        memset(root,sizeof(root),);
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++){
            scanf("%d",&a[i]);
            dctz[i].sum=a[i];
            dctz[i].num=i;
        }
        sort(dctz+,dctz+n+,cmp);
        for(int i=;i<=n;i++)a[dctz[i].num]=i,b[i]=dctz[i].sum;
        tot=;
        root[]=;
        tr[].sum=;
        build(,,n);
        for(int i=;i<=n;i++){
            root[i]=tot+;
            insert(root[i-],a[i],,n);
        }
        for(int i=;i<=m;i++){
            scanf("%d%d%d",&l,&r,&k);
            int x=search(k,root[l-],root[r],,n);
            printf("%d\n",b[x]);
        }
    }
}