【BZOJ3123】森林（主席树，启发式合并）

题面

神TM题面是图片

题解

首先，求树链上第k大，请参看BZOJ2588 Count On a Tree
这道题目于是增添了一个动态的合并森林的操作
所以直接启发式合并就可以啦
我第一次交直接T了
加了一堆rg就AC了。。。
神奇的register

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 81000
#define rg register
inline int read()
{
    rg int x=0,t=1;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line
{
    int v,next;
}e[MAX<<1];
int h[MAX],cnt=1;
int p[MAX][20];
inline void Add(int u,int v)
{
    e[cnt]=(Line){v,h[u]};
    h[u]=cnt++;
}
struct Node
{
    int ls,rs;
    int v;
}t[MAX<<9];
int tot,n,T,m;
int fa[MAX],dep[MAX],rt[MAX],sum,S[MAX];
int a[MAX];
void Modify(int &now,int ff,int l,int r,int pos)
{
    now=++tot;
    t[now]=t[ff];
    t[now].v++;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(pos<=mid)Modify(t[now].ls,t[ff].ls,l,mid,pos);
    else Modify(t[now].rs,t[ff].rs,mid+1,r,pos);
}
void Build(int &now,int l,int r)
{
    now=++tot;
    if(l==r)return;
    rg int mid=(l+r)>>1;
    Build(t[now].ls,l,mid);
    Build(t[now].rs,mid+1,r);
}
void dfs(int u,int ff)
{
    fa[u]=p[u][0]=ff;dep[u]=dep[ff]+1;
    Modify(rt[u],rt[ff],1,sum,a[u]);
    for(int i=1;i<18;++i)p[u][i]=p[p[u][i-1]][i-1];
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==ff)continue;
        dfs(v,u);
    }
}
int Query(int A,int B,int C,int D,int l,int r,int k)
{
    if(l==r)return l;
    rg int tmp=t[t[A].ls].v+t[t[B].ls].v-t[t[C].ls].v-t[t[D].ls].v;
    rg int mid=(l+r)>>1;
    if(k<=tmp)return Query(t[A].ls,t[B].ls,t[C].ls,t[D].ls,l,mid,k);
    else return Query(t[A].rs,t[B].rs,t[C].rs,t[D].rs,mid+1,r,k-tmp);
}
int f[MAX],size[MAX];
int getf(int x){return x==f[x]?x:f[x]=getf(f[x]);}
void merge(int x,int y){x=getf(x);y=getf(y);f[y]=x;size[x]+=size[y];size[y]=0;}
inline int LCA(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=17;i>=0;--i)
        if(dep[p[x][i]]>=dep[y])x=p[x][i];
    if(x==y)return x;
    for(int i=17;i>=0;--i)
        if(p[x][i]!=p[y][i])
            x=p[x][i],y=p[y][i];
    return p[x][0];
}
int main()
{
    read();
    n=read();m=read();T=read();
    for(rg int i=1;i<=n;++i)S[++sum]=a[i]=read(),size[i]=1;
    sort(&S[1],&S[sum+1]);
    sum=unique(&S[1],&S[sum+1])-S-1;
    for(rg int i=1;i<=n;++i)a[i]=lower_bound(&S[1],&S[sum+1],a[i])-S;
    Build(rt[0],1,sum);
    for(rg int i=1;i<=m;++i)
    {
        rg int x=read(),y=read();
        Add(x,y);Add(y,x);
        if(size[getf(x)]>size[getf(y)])merge(x,y);
        else merge(y,x);
    }
    for(rg int i=1;i<=n;++i)
    {
        if(!rt[i])
            dfs(i,0);
    }
    rg char ch[20];
    rg int ls=0;
    while(T--)
    {
        scanf("%s",ch);
        if(ch[0]=='Q')
        {
            rg int u=read()^ls,v=read()^ls,k=read()^ls;
            rg int lca=LCA(u,v);
            printf("%d\n",ls=S[Query(rt[u],rt[v],rt[lca],rt[p[lca][0]],1,sum,k)]);
        }
        else
        {
            rg int x=read()^ls,y=read()^ls;
            if(size[getf(x)]>size[getf(y)])swap(x,y);
            merge(y,x);
            dfs(x,y);
            Add(x,y);Add(y,x);
        }
    }
    return 0;
}