P3302 [SDOI2013]森林 （树上主席树+启发式合并）

题目链接

题面：

题解：
路径上第 k 大用树上主席树完成。
合并的时候直接小树往大树上合并（启发式合并）。
时间复杂度大概O（n logn logn），也不太会算。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#define ll long long
#define llu unsigned ll
#define pr make_pair
#define pb push_back
#define y1 yy
using namespace std;
const int maxn=80100;
int head[maxn],ver[maxn<<1],nt[maxn<<1],vi[maxn],d[maxn],b[maxn];
int f[maxn][20],si[maxn],in[maxn],nowt,tot,cnt,n,m,tt,cm;
char str[12];
struct node
{
    int lc,rc;
    int sum;
}t[maxn*400];
int root[maxn];


void add(int x,int y)
{
    ver[++tot]=y,nt[tot]=head[x],head[x]=tot;
}

int _insert(int now,int pos,int l,int r)
{
    int p=++cnt;
    t[p]=t[now];
    if(l==r)
    {
        t[p].sum++;
        return p;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) t[p].lc=_insert(t[now].lc,pos,l,mid);
    else t[p].rc=_insert(t[now].rc,pos,mid+1,r);
    t[p].sum=t[t[p].lc].sum+t[t[p].rc].sum;
    return p;
}

int ask(int fafa,int fa,int x,int y,int l,int r,int k)
{
    if(l==r) return l;
    int mid=(l+r)>>1;
    int pm=t[t[x].lc].sum+t[t[y].lc].sum-t[t[fa].lc].sum-t[t[fafa].lc].sum;
    if(pm>=k) return ask(t[fafa].lc,t[fa].lc,t[x].lc,t[y].lc,l,mid,k);
    else return ask(t[fafa].rc,t[fa].rc,t[x].rc,t[y].rc,mid+1,r,k-pm);
}

void dfs(int x,int fa,int now)
{
    in[x]=now;
    si[x]=1;
    root[x]=_insert(root[fa],vi[x],1,cm);
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        if(y==fa) continue;
        d[y]=d[x]+1;
        f[y][0]=x;
        for(int j=1;j<=nowt;j++)
            f[y][j]=f[f[y][j-1]][j-1];
        dfs(y,x,now);
        si[x]+=si[y];
    }
}

int lca(int x,int y)
{
    if(d[x]>d[y]) swap(x,y);
    for(int i=nowt;i>=0;i--)
        if(d[f[y][i]]>=d[x]) y=f[y][i];
    if(x==y) return x;
    for(int i=nowt;i>=0;i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}

int main(void)
{
    int nn;
    scanf("%d",&nn);

    int x,y;
    scanf("%d%d%d",&n,&m,&tt);
    nowt=log(n)/log(2)+1;

    for(int i=1;i<=n;i++)
        scanf("%d",&vi[i]),b[i]=vi[i];
    sort(b+1,b+n+1);
    cm=unique(b+1,b+n+1)-(b+1);
    for(int i=1;i<=n;i++)
        vi[i]=lower_bound(b+1,b+cm+1,vi[i])-b;

    for(int i=1;i<=m;i++)
        scanf("%d%d",&x,&y),add(x,y),add(y,x);

    for(int i=1;i<=n;i++)
    {
        if(!d[i])
        {
            d[i]=1;
            dfs(i,0,i);
        }
    }

    int last=0;
    for(int i=1;i<=tt;i++)
    {
        scanf("%s",&str);
        if(str[0]=='Q')
        {
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            x^=last,y^=last,k^=last;
            int la=lca(x,y);
            last=ask(root[f[la][0]],root[la],root[x],root[y],1,cm,k);
            printf("%d\n",last=b[last]);
        }
        else
        {
            scanf("%d%d",&x,&y);
            x^=last,y^=last;
            add(x,y),add(y,x);
            int xfa=in[x],yfa=in[y];
            if(si[xfa]>si[yfa])
            {
                swap(xfa,yfa);
                swap(x,y);
            }
            si[yfa]+=si[xfa];
            f[x][0]=y;d[x]=d[y]+1;
            for(int i=1;i<=nowt;i++)
                f[x][i]=f[f[x][i-1]][i-1];
            dfs(x,y,yfa);
        }
    }
    return 0;

}