主席树学习总结

主席树即每次建立改变点的链接，建立n颗权值线段树，根据其类似于前缀和的性质，求解区间第k大问题。

POJ - 2104 数组区间上求第k大：

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<stack>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=1e5+100;
int ls[maxn*20],rs[maxn*20],root[maxn*20],sum[maxn*20];
int tot;
void build(int &rt,int l,int r)
{
    rt=++tot;
    sum[rt]=0;
    if(l==r)
        return;
    int mid=l+r>>1;
    build(ls[rt],l,mid);
    build(rs[rt],mid+1,r);
}
void update(int &rt,int l,int r,int last,int p)
{
    rt=++tot;
    ls[rt]=ls[last];
    rs[rt]=rs[last];
    sum[rt]=sum[last]+1;
    if(l==r)
        return;
    int mid=l+r>>1;
    if(p<=mid)
        update(ls[rt],l,mid,ls[last],p);
    else
        update(rs[rt],mid+1,r,rs[last],p);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r)return l;
    int mid=l+r>>1;
    int cnt=sum[ls[y]]-sum[ls[x]];
    if(k<=cnt)
        return query(l,mid,ls[x],ls[y],k);
    else
        return query(mid+1,r,rs[x],rs[y],k-cnt);
}
int a[maxn],num[maxn];
int main()
{
    int n,m;
     scanf("%d%d",&n,&m);
    tot=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        num[i]=a[i];
    }
    sort(a+1,a+1+n);
    int cnt=unique(a+1,a+1+n)-a-1;
    build(root[0],1,cnt);
    for(int i=1;i<=n;i++)
        num[i]=lower_bound(a+1,a+1+cnt,num[i])-a;
    for(int i=1;i<=n;i++)
        update(root[i],1,cnt,root[i-1],num[i]);
    int l,r,k;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&l,&r,&k);
        int ans=query(1,cnt,root[l-1],root[r],k);
        printf("%d\n",a[ans]);
    }
    return 0;
}

SPOJ - COT树上两点u到v之间权值求第k大，lca+主席树

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5;
int pre[maxn];
int dep[maxn];
int f[maxn][25];
int n,m;
vector<int>v[maxn];
int root[maxn*20];
int sum[maxn*20];
int ls[maxn*20];
int rs[maxn*20];
int tot,cnt;
int a[maxn],num[maxn];
void update(int &rt,int l,int r,int last,int p)
{
    rt=++tot;
    ls[rt]=ls[last];
    rs[rt]=rs[last];
    sum[rt]=sum[last]+1;
    if(l==r)
        return ;
    int mid=l+r>>1;
    if(p<=mid)
        update(ls[rt],l,mid,ls[last],p);
    else
        update(rs[rt],mid+1,r,rs[last],p);
}
void dfs(int now,int fa)
{
    pre[now]=fa;
    dep[now]=dep[fa]+1;
    f[now][0]=fa;
    for(int i=1;i<20;i++)
    {
        f[now][i]=f[f[now][i-1]][i-1];
    }
    update(root[now],1,cnt,root[fa],num[now]);
    for(int i=0;i<v[now].size();i++){
        if(v[now][i]!=fa)
            dfs(v[now][i],now);
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=19;i>=0;i--)
    {
        if(dep[x]-(1<<i)>=dep[y])
            x=f[x][i];
    }
    if(x==y)
        return x;
    for(int i=19;i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    }
    return f[x][0];
}
int query(int l,int r,int x,int y,int z,int fz,int k)
{
    if(l==r)
        return l;
    int mid=l+r>>1;
    int ssum=sum[ls[x]]+sum[ls[y]]-sum[ls[z]]-sum[ls[fz]];
    if(k<=ssum)
        return query(l,mid,ls[x],ls[y],ls[z],ls[fz],k);
    else
        return query(mid+1,r,rs[x],rs[y],rs[z],rs[fz],k-ssum);
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(root,0,sizeof(root));
        memset(ls,0,sizeof(ls));
        memset(rs,0,sizeof(rs));
        memset(sum,0,sizeof(sum));
        memset(dep,0,sizeof(dep));
        memset(f,0,sizeof(f));
        tot=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            num[i]=a[i];
            v[i].clear();
        }
        sort(a+1,a+1+n);
        cnt=unique(a+1,a+1+n)-a-1;
        for(int i=1;i<=n;i++)
            num[i]=lower_bound(a+1,a+1+cnt,num[i])-a;
        int x,y;
        for(int i=1;i<=n-1;i++)
        {
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        dfs(1,0);
        int l,r,k;
        while(m--)
        {
            scanf("%d%d%d",&l,&r,&k);
            int lcaa=lca(l,r);
            int ans=query(1,cnt,root[l],root[r],root[lcaa],root[pre[lcaa]],k);
            printf("%d\n",a[ans]);
        }
    }
    return 0;
}