bzoj千题计划258：bzoj3123: [Sdoi2013]森林

http://www.lydsy.com/JudgeOnline/problem.php?id=3123

启发式合并主席树

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

#define N 80001

int n,testcase,m,T;
int val[N];
int front[N],to[N<<],nxt[N<<],tot;

int has[N],all;

int fa[N][],dep[N],lim;

int F[N],siz[N];

int id,root[N];

struct node
{
    int lc,rc,cnt;
}tr[N*];

void read(int &x)
{
    x=; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*+c-''; c=getchar(); }
}

void add(int u,int v)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot;
}

int find(int i) { return F[i]==i ? i : F[i]=find(F[i]); }

void unionn(int u,int v)
{
    u=find(u); v=find(v);
    siz[u]+=siz[v];
    F[v]=u;
}

void init()
{
    read(testcase);
    read(n); read(m); read(T);
    lim=log(n)/log();
    for(int i=;i<=n;++i)
    {
        read(val[i]);
        F[i]=i;
        siz[i]=;
    }
    int u,v;
    for(int i=;i<=m;++i)
    {
        read(u); read(v);
        add(u,v);
        unionn(u,v);
    }
}
void insert(int &x,int y,int l,int r,int pos)
{
    if(!x) x=++id;
    tr[x].cnt=tr[y].cnt+;
    if(l==r) return;
    int mid=l+r>>;
    if(pos<=mid)
    {
        tr[x].rc=tr[y].rc;
        insert(tr[x].lc,tr[y].lc,l,mid,pos);
    }
    else
    {
        tr[x].lc=tr[y].lc;
        insert(tr[x].rc,tr[y].rc,mid+,r,pos);
    }
}

int query(int u,int v,int lca,int flca,int l,int r,int k)
{
    if(l==r) return l;
    int mid=l+r>>,tmp=tr[tr[u].lc].cnt+tr[tr[v].lc].cnt-tr[tr[lca].lc].cnt-tr[tr[flca].lc].cnt;
    if(k<=tmp) return query(tr[u].lc,tr[v].lc,tr[lca].lc,tr[flca].lc,l,mid,k);
               return query(tr[u].rc,tr[v].rc,tr[lca].rc,tr[flca].rc,mid+,r,k-tmp);
}

void discrete()
{
    for(int i=;i<=n;++i) has[i]=val[i];
    sort(has+,has+n+);
    all=unique(has+,has+n+)-has-;
    for(int i=;i<=n;++i) val[i]=lower_bound(has+,has+all+,val[i])-has;
}    

void dfs(int x)
{
    dep[x]=dep[fa[x][]]+;
    insert(root[x],root[fa[x][]],,all,val[x]);
    for(int i=;i<=lim;++i) fa[x][i]=fa[fa[x][i-]][i-];
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=fa[x][])
        {
            fa[to[i]][]=x;
            dfs(to[i]);
        }
}

int get_lca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    int x=dep[u]-dep[v];
    for(int i=lim;i>=;--i)
        if(x&(<<i)) u=fa[u][i];
    if(u==v) return u;
    for(int i=lim;i>=;--i)
        if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
    return fa[u][];
}

void solve()
{
    for(int i=;i<=n;++i)
        if(!fa[i][]) dfs(i);
    char ch[];
    int u,v,k,last=;
    int lca,fu,fv;
    while(T--)
    {
        scanf("%s",ch);
        if(ch[]=='Q')
        {
            read(u); read(v); read(k);
            u^=last; v^=last; k^=last;
            lca=get_lca(u,v);
            last=has[query(root[u],root[v],root[lca],root[fa[lca][]],,all,k)];
            printf("%d\n",last);
        }
        else
        {
            read(u); read(v);
            u^=last; v^=last;
            fu=find(u); fv=find(v);
            if(siz[fu]<siz[fv]) swap(fu,fv);
            unionn(u,v);
            add(u,v);
            fa[v][]=u;
            dfs(v);
        }
    }
}

int main()
{
    init();
    discrete();
    solve();
}