P3348 [ZJOI2016]大森林（LCT）

对于每个1操作建一个虚点，以后的0操作都连在最近建好的虚点上。这样每次整体嫁接的时候，直接把这个虚点断掉它原来的父亲，再link过去就可以了

求在x位置的两点之间距离,只要之前的换点加点操作完成,就可以计算,而且也要马上计算,之后树的形态就又要变了,这样保证了复杂度

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define Debug(x) cout<<#x<<"="<<x<<endl
using namespace std;
typedef long long LL;
const int INF=1e9+7;
inline LL read(){
    register LL x=0,f=1;register char c=getchar();
    while(c<48||c>57){if(c=='-')f=-1;c=getchar();}
    while(c>=48&&c<=57)x=(x<<3)+(x<<1)+(c&15),c=getchar();
    return f*x;
}

const int MAXN=3e5+5;
const int MAXM=2e5+5;

struct Query{
    int pos,id,x,y;
    inline friend bool operator<(Query a,Query b){
        if(a.pos==b.pos) return a.id<b.id;
        return a.pos<b.pos;
    }
}q[MAXM];
int Qcnt,Acnt;

int at[MAXN],L[MAXN],R[MAXN],ans[MAXN];
int n,m,p,aux,real;

namespace LCT{
    int ch[MAXN][2],fa[MAXN],size[MAXN],val[MAXN];
    int st[MAXN],top;
#define ls (ch[rt][0])
#define rs (ch[rt][1])
    inline bool chk(int x){return ch[fa[x]][1]==x;}
    inline bool isnotroot(int x){return ch[fa[x]][0]==x||ch[fa[x]][1]==x;}
    inline void pushup(int rt){size[rt]=size[ls]+size[rs]+val[rt];}//只要算实点
    inline void rotate(int x){
        int y=fa[x],z=fa[y],k=chk(x),w=ch[x][k^1];
        ch[y][k]=w,fa[w]=y;
        if(isnotroot(y)) ch[z][chk(y)]=x; fa[x]=z;
        ch[x][k^1]=y,fa[y]=x;
        pushup(y);pushup(x);
    }
    inline void splay(int x){
        while(isnotroot(x)){
            int y=fa[x];
            if(isnotroot(y)){
                if(chk(x)==chk(y)) rotate(y);
                else rotate(x);
            }
            rotate(x);
        }
    }
    inline int access(int x){
        int y=0;
        for(;x;x=fa[y=x])
            splay(x),ch[x][1]=y,pushup(x);
        return y;//access(y)时最后跳的虚边的父亲就是lca,即最后的y
    }
    inline void link(int x,int y){
        splay(x);//不能makeroot,只能splay
        fa[x]=y;//只有根节点才能连边
    }
    inline void cut(int x){//和它上面的点断开
        access(x);splay(x);
        ch[x][0]=fa[ch[x][0]]=0;
        pushup(x);
    }
#undef ls
#undef rs
}using namespace LCT;

int main(){
    n=read(),m=read();
    real=1,size[1]=1,val[1]=1,at[1]=1,L[1]=1,R[1]=n;//初始时只有1一个实点
    link(p=aux=2,1);
    for(int i=1;i<=m;i++){
        int op=read(),x=read(),y=read();
        if(op==0){
            link(at[++real]=++p,aux);///新建一个点连到虚点上去,并记录第real个实点在哪
            size[p]=1,val[p]=1;//更新val[]!!!
            L[real]=x,R[real]=y;
        }
        if(op==1){
            int t=read();
            x=max(x,L[t]),y=min(y,R[t]);//有些树还没这个点
            if(x>y) continue;
            link(++p,aux);///新建一个点连到虚点上去
            //size[p]=0,val[p]=0; 虚点没有val[],不算进路径上有size[]个点
            q[++Qcnt]=(Query){x,i-m,p,at[t]};/// -m是为了使换点操作在加点操作前面.
            q[++Qcnt]=(Query){y+1,i-m,p,aux};
            aux=p;//更换虚点了!!!
        }
        if(op==2) q[++Qcnt]=(Query){x,++Acnt,at[y],at[read()]};//在x位置的两点之间距离,只要之前的换点加点操作完成,就可以计算,而且也要马上计算,之后树的形态就又要变了
    }
    sort(q+1,q+Qcnt+1);
    for(int i=1;i<=Qcnt;i++){
        if(q[i].id>0){
            int x=q[i].x,y=q[i].y,&sum=ans[q[i].id];
            //由于根节点始终为1,不能makeroot后求路径,所以要用到lca.本题中步数=size[x]+size[y]-2*size[lca].
            access(x);splay(x);sum+=size[x];
            int lca=access(y);splay(y);sum+=size[y];
            access(lca);splay(lca);sum-=size[lca]*2;
        }
        else{
            cut(q[i].x);//和它上面的点断开
            link(q[i].x,q[i].y);//然后连到新的点上去
        }
    }
    for(int i=1;i<=Acnt;i++)
        printf("%d\n",ans[i]);
}

P3348 [ZJOI2016]大森林（LCT）

猜你喜欢