对Kth number的一点小研究——主席树

显然网上已经有太多的人写过这道题的题解了，而且大家都写得挺详细的，这里我就分析一份代码，没学过的同学在看了别人的博客之后看这个代码分析，应该可以很快解决主席树的基础学习

---

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<set>
using namespace std;
const int MAXN = 100010;
struct Node{
    int L,R,sum;
};
Node T[MAXN*20];
int timer,root[MAXN],n,m,rank[MAXN],ll,rr,kk;
struct Line{
    int x,idx;
}a[MAXN];
bool cmp(const Line& A,const Line& B){return A.x<B.x;}
void insert(int num,int &x,int l,int r){新的节点为x,之前只是给root赋了值,但是并没有给空间,所以我们下一句话就给空间
    T[++timer]=T[x];x=timer;++T[x].sum;//这里的新的空间开给x这个
    if(l==r) return;//如果l==r,那么已经更新到叶节点了,返回
    int mid=(l+r)>>1;
    if(num<=mid)insert(num,T[x].L,l,mid);//往作或者往右赋值,注意线段树存的是第几个数,而不是数
    else        insert(num,T[x].R,mid+1,r);
}
int query(int x,int y,int k,int l,int r){
    if(l==r) return l;//返回位置
    int mid=(l+r)>>1;
    int t=T[T[y].L].sum-T[T[x].L].sum;//版本y的主席树减去版本x的主席树的左区间,也就是说,在y这个版本和x这个版本之间,左区间到底插入了多少数的意思
    if(k<=t) return query(T[x].L,T[y].L,k,l,mid);//分别递归左右区间
    else     return query(T[x].R,T[y].R,k-t,mid+1,r);
}
int main(){
    T[0].L=T[0].R=T[0].sum=0;
    root[0]=0;//首先对建一棵0节点的空树
    while(scanf("%d%d",&n,&m)!=EOF){
        for(register int i=1;i<=n;i++){
            scanf("%d",&a[i].x);
            a[i].idx=i;//读入,idx是指i这一位的数组下标
        }
        sort(a+1,a+n+1,cmp);//按照x关键字从小到大排序
        for(register int i=1;i<=n;i++) rank[a[i].idx]=i;
        timer=1;//rank表示原来的第idx位,现在的标号是i
        for(register int i=1;i<=n;i++){
            root[i]=root[i-1];//先赋值一棵树过去
            insert(rank[i],root[i],1,n);//然后再插入,插入的时候,用rank[i]插入,原因是按照最开始的插入顺序的排序之后的位置进行插入,相当于离散化了,真的是非常厉害,然后新构建root[i]这棵树
        }
        while(m--){
            scanf("%d%d%d",&ll,&rr,&kk);
            printf("%d\n",a[query(root[ll-1],root[rr],kk,1,n)].x); //函数式线段树支持两棵线段树的加减,让我们来看看query函数
        }
    }
    return 0;
}

---

注：
代码来源于 http://www.cnblogs.com/oyking/p/3230296.html
我自己敲了一遍 OTZ oyking %%%