poj2104 主席树区间第k大

主席树裸题，过了好久都快忘了。

主席树主要就是使多个线段树并行于一颗树中，保存的是一段区间内数字出现的个数，所以先离散化。

对整个序列构造一个线段树，然后在线段树上不断找第 k 小在当前数字区间的左半部分还是右半部分。（类似于平衡树）

数字的更改只影响了一条从这个叶子节点到根的路径，所以只有这条路径是新的，其他的都没有改变。比如对于某个节点，要往右边走，那么左边那些就不用新建，只要用个指针链到原树的此节点左边就可以了

见代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100100;
struct node
{
    int l,r,sum;
} T[maxn*40];
int n,m,cnt,root[maxn],x,y,k;
int a[maxn],b[maxn],p[maxn];
bool cmp(const int &i,const int &j)
{
    return a[i]<a[j];
}
void update(int l,int r,int &x,int &y,int pos)
{
    T[++cnt]=T[y],T[cnt].sum++,x=cnt;
    if(l==r) return;
    int mid=(l+r)/2;
    if(mid>=pos) update(l,mid,T[x].l,T[y].l,pos);
    else update(mid+1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r) return l;
    int mid=(l+r)/2;
    int sum=T[T[y].l].sum-T[T[x].l].sum;
    if(sum>=k) return query(l,mid,T[x].l,T[y].l,k);
    else return query(mid+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=i;
    sort(b+1,b+1+n,cmp);
    for(int i=1;i<=n;i++) p[b[i]]=i;
    for(int i=1;i<=n;i++) update(1,n,root[i],root[i-1],p[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",a[b[query(1,n,ro``````
t[x-1],root[y],k)]]); 
    }
    return 0;
}