主席树 介绍

转：http://blog.csdn.net/metalseed/article/details/8045038

先根据 Seter 留下的文本做一些记录。。

主席树大概是一种离线结构，我以前反正没看到过这东西，所以就自己给他起名字了！如果谁知道这东西的真名，请告诉我！

现在我们知道，主席树的全名应该是 函数式版本的线段树。加上附带的一堆 technology。。
。。总之由于原名字太长了，而且 “主席” 两个字念起来冷艳高贵，以后全部称之为主席树好了。。。

主席树的主体是线段树，准确的说，是很多棵线段树，存的是一段数字区间出现次数（所以要先离散化可能出现的数字）。举个例子，假设我每次都要求整个序列内的第 k 小，那么对整个序列构造一个线段树，然后在线段树上不断找第 k 小在当前数字区间的左半部分还是右半部分。这个操作和平衡树的 Rank 操作一样，只是这里将离散的数字搞成了连续的数字。

先假设没有修改操作：

对于每个前缀 S1…i，保存这样一个线段树 Ti，组成主席树。这样不是会 MLE 么？最后再讲。

注意，这个线段树对一条线段，保存的是这个数字区间的出现次数，所以是可以互相加减的！还有，由于每棵线段树都要保存同样的数字，所以它们的大小、形态也都是一样的！这实在是两个非常好的性质，是平衡树所不具备的。

对于询问 (i,j)，我只要拿出 Tj 和 Ti-1，对每个节点相减就可以了。说的通俗一点，询问 i..j 区间中，一个数字区间的出现次数时，就是这些数字在 Tj 中出现的次数减去在 Ti-1 中出现的次数。

那么有修改操作怎么办呢？

如果将询问看成求一段序列的数字和，那么上面那个相当于求出了前缀和。加入修改操作后，就要用树状数组等来维护前缀和了。于是那个 “很好的性质” 又一次发挥了作用，由于主席树可以互相加减，所以可以用树状数组来套上它。做法和维护前缀和长得基本一样，不说了。

这段指出了主席树的主要性质。。

线段树的每个结点，保存的是这个区间含有的数字的个数。
主席树的每个结点，也就是每颗线段树的大小和形态也是一样的，也就是主席树之间可以相互进行加减运算。。
同时我们也枚举一下主席树的一些局限：

主席树是一种离线结构。。（必须预先知道所有数字的范围。。这在一些应用中会成为障碍。。。
存在 MLE 问题。。（如果按照定义里面的方法去写，对每个结点都需要开一整可线段树，至少都是 O(n2) 级别的空间。。
——————————————————
开始填坑。由于每棵线段树的大小形态都是一样的，而且初始值全都是 0，那每个线段树都初始化不是太浪费了？所以一开始只要建一棵空树即可。

然后是在某棵树上修改一个数字，由于和其他树相关联，所以不能在原来的树上改，必须弄个新的出来。难道要弄一棵新树？不是的，由于一个数字的更改只影响了一条从这个叶子节点到根的路径，所以只要只有这条路径是新的，另外都没有改变。比如对于某个节点，要往右边走，那么左边那些就不用新建，只要用个指针链到原树的此节点左边就可以了，这个步骤的前提也是线段树的形态一样。

假设s是数字个数，这个步骤的空间复杂度显然是 O(logs)。用树状数组去套它，共有 2logn 棵树被修改，m 个操作再加上一开始的空树和 n 个数字，总共就是 O((n+m)lognlogs)。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int maxn=1e5+10;

struct seg
{
    int lson,rson,num;
}T[maxn*40];
int a[maxn],root[maxn];
vector<int> v;
int cnt=0;
int getid(int x) { return lower_bound(v.begin(),v.end(),x)-v.begin();}
void update(int l,int r,int &now,int pre,int a)
{
    now=++cnt;
    T[now]=T[pre];
    T[now].num++;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(a<=mid) update(l,mid,T[now].lson,T[pre].lson,a);
    else update(mid+1,r,T[now].rson,T[pre].rson,a);
}

int query(int l,int r,int pre,int now,int k)
{
    if(l==r) return l;
    int mid=(l+r)>>1;
    int lnum=T[T[now].lson].num-T[T[pre].lson].num;
    if(k<=lnum)
    {
        return query(l,mid,T[pre].lson,T[now].lson,k);
    }
    else return query(mid+1,r,T[pre].rson,T[now].rson,k-lnum);
}

int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1;i<=n;i++)
        update(0,n,root[i],root[i-1],getid(a[i]));
    int a,b,k;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&a,&b,&k);
        printf("%d\n",v[query(0,n,root[a-1],root[b],k)] );
    }
}