BZOJ 3123 SDOI2013森林

Problem

BZOJ

Solution

刚开始一看题目，竟然有连边操作，还强制在线，当时就懵逼了，难不成要lct套主席树，感觉很不可做

那我们缩一缩树上主席树吧，这年头连树都要上树了，不过感觉主席树的题不需要动不动就链剖了23333
同样也是前缀的思想，树上的前缀就是从根到节点路径的前缀和了。
于是乎，对于一条路径在树上的答案，我们需要四个参数x,y,lca,f[lca]，答案即为sum[x]+sum[y]-sum[lca]-sum[f[lca]]。这是显然的，画个图就可以知道。然后就递归查询即可出答案。
因此，我们建树的复杂度应该是$O(nlogn)$

除此之外，我们考虑连边操作。我们需要合并两棵树，则我们必须对其中一棵树重新dfs建立主席树，复杂度同上，考虑启发式合并。所以总体的时间复杂度为$O(nlog^2n)$。

Code

#include <algorithm>
#include <cstdio>
using namespace std;
const int maxn=80010,maxm=10000010;
struct data{int v,nxt;}edge[maxn<<1];
int n,m,q,p,len,tot,lst,a[maxn],b[maxn],head[maxn],f[maxn][20],sz[maxn],deep[maxn];
int rt[maxn],sum[maxm],lc[maxm],rc[maxm];
char op[5];
inline int getrt(int x){for(int i=19;~i;i--) if(f[x][i]) x=f[x][i];return x;}
inline void insert(int u,int v)
{
    edge[++p]=(data){v,head[u]};head[u]=p;
    edge[++p]=(data){u,head[v]};head[v]=p;
}
void input()
{
    int x,y;
    scanf("%d",&n);scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
    sort(b+1,b+n+1);
    len=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+len+1,a[i])-b;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        insert(x,y);
    }
}
void update(int l,int r,int pos,int& rt)
{
    tot++;lc[tot]=lc[rt];rc[tot]=rc[rt];
    sum[tot]=sum[rt]+1;rt=tot;
    if(l==r) return ;
    int m=(l+r)>>1;
    if(pos<=m) update(l,m,pos,lc[rt]);
    else update(m+1,r,pos,rc[rt]);
}
int query(int l,int r,int x,int y,int lca,int flca,int k)
{
    if(l==r) return l;
    int m=(l+r)>>1,cnt=0;
    cnt=sum[lc[x]]+sum[lc[y]]-sum[lc[lca]]-sum[lc[flca]];
    if(k<=cnt) return query(l,m,lc[x],lc[y],lc[lca],lc[flca],k);
    else return query(m+1,r,rc[x],rc[y],rc[lca],rc[flca],k-cnt);
}
void dfs(int x)
{
    for(int i=1;i<20;i++)
      f[x][i]=f[f[x][i-1]][i-1];
    rt[x]=rt[f[x][0]];sz[x]=1;
    update(1,len,a[x],rt[x]);
    for(int i=head[x];i;i=edge[i].nxt)
      if(edge[i].v!=f[x][0])
      {
        f[edge[i].v][0]=x;deep[edge[i].v]=deep[x]+1;
        dfs(edge[i].v);sz[x]+=sz[edge[i].v];
      }
}
int getlca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    int t=deep[x]-deep[y];
    for(int i=0;i<20;i++)
      if(t&(1<<i))
        x=f[x][i];
    if(x==y) return x;
    for(int i=19;~i;i--)
      if(f[x][i]!=f[y][i])
        x=f[x][i],y=f[y][i];
    return f[x][0];
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    input();
    for(int i=1;i<=n;i++)
      if(!f[i][0])
        dfs(i);
    for(int i=1,x,y,k;i<=q;i++)
    {
        scanf("%s",op);
        if(op[0]=='L')
        {
            scanf("%d%d",&x,&y);x^=lst;y^=lst;
            int fx=getrt(x),fy=getrt(y);insert(x,y);
            if(sz[fx]>sz[fy]) swap(fx,fy),swap(x,y);
            f[x][0]=y;deep[x]=deep[y]+1;sz[fy]+=sz[fx];
            dfs(x);
        }
        else if(op[0]=='Q')
        {
            scanf("%d%d%d",&x,&y,&k);x^=lst;y^=lst;k^=lst;
            int lca=getlca(x,y);
            printf("%d\n",lst=b[query(1,len,rt[x],rt[y],rt[lca],rt[f[lca][0]],k)]);
        }
    }
    return 0;
}