bzoj4530 [BJOI2014]大融合

http://www.elijahqi.win/2018/02/12/bzoj4530/
Description
小强要在N个孤立的星球上建立起一套通信系统。这套通信系统就是连接N个点的一个树。
这个树的边是一条一条添加上去的。在某个时刻，一条边的负载就是它所在的当前能够
联通的树上路过它的简单路径的数量。
例如，在上图中，现在一共有了5条边。其中，(3,8)这条边的负载是6，因
为有六条简单路径2-3-8,2-3-8-7,3-8,3-8-7,4-3-8,4-3-8-7路过了(3,8)。
现在，你的任务就是随着边的添加，动态的回答小强对于某些边的负载的
询问。
Input
第一行包含两个整数N,Q，表示星球的数量和操作的数量。星球从1开始编号。
接下来的Q行，每行是如下两种格式之一：
A x y 表示在x和y之间连一条边。保证之前x和y是不联通的。
Q x y 表示询问(x,y)这条边上的负载。保证x和y之间有一条边。
1≤N,Q≤100000
Output
对每个查询操作，输出被查询的边的负载。
Sample Input

8 6
A 2 3
A 3 4
A 3 8
A 8 7
A 6 5
Q 3 8
Sample Output

6
HINT

Source

鸣谢佚名上传

昨天思考了很久 今天早晨起来给实现一下 这题只需要我同时维护一下虚边子树的信息即可 注意维护的顺序 为了避免重复 定义sizex为这个节点所有虚子树的信息 size为这个节点所有的子树信息 那么接下来只需要在access操作和link操作改一改就Okay 但是注意在link操作的时候我需要同时把两棵子树变成根 否则对于另一棵子树我这个节点上方的虚边的信息就没法维护 查询答案的时候直接把两个节点的节点数乘起来即可

#include<cstdio>
#include<algorithm>
#define N 110000
#define ll long long
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=gc();}
    while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int fa[N],size[N],sizex[N],c[N][2],rev[N],n,m,q[N],top;
inline bool isroot(int x){return x!=c[fa[x]][0]&&x!=c[fa[x]][1];}
inline void update(int x){
    size[x]=size[c[x][0]]+size[c[x][1]]+sizex[x]+1;
}
inline void pushdown(int x){
    if (!rev[x]) return;rev[x]=0;
    int l=c[x][0],r=c[x][1];rev[l]^=1;
    swap(c[x][0],c[x][1]);rev[r]^=1;
}
inline void rotate(int x){
    int y=fa[x],z=fa[y];
    if(!isroot(y)) c[z][c[z][1]==y]=x;
    int l=c[y][1]==x,r=l^1;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x){
    q[top=1]=x;for (int i=x;!isroot(i);i=fa[i]) q[++top]=fa[i];
    while(top) pushdown(q[top--]);
    while(!isroot(x)){
        int y=fa[x],z=fa[y];
        if(!isroot(y))
            if (c[y][0]==x^c[z][0]==y) rotate(x);else rotate(y);
        rotate(x);
    }
}
inline void access(int x){
    for (int t=0;x;t=x,x=fa[x]) splay(x),sizex[x]+=size[c[x][1]]-size[t],c[x][1]=t,update(x);
}
inline void makeroot(int x){
    access(x);splay(x);rev[x]^=1;
}
inline void print(int x){
    pushdown(x);
    if (c[x][0]) print(c[x][0]);
    printf("%d %d %d %d %d\n",x,size[x],sizex[x],c[x][0],c[x][1]);
    if (c[x][1]) print(c[x][1]);
}
int main(){
    freopen("bzoj4530.in","r",stdin);
    n=read();m=read();
    for (int i=1;i<=n;++i) size[i]=1;
    while(m--){
        char ch=gc();while(ch!='A'&&ch!='Q') ch=gc();
        if (ch=='A'){
            int x=read(),y=read();
            makeroot(x);makeroot(y);fa[x]=y;sizex[y]+=size[x];update(y);//puts("this is x");print(x);puts("this is y");print(y);
            //printf("%d\n",m);

        }
        if (ch=='Q'){
            int x=read(),y=read();
            makeroot(x);access(y);splay(y);//puts("this is x");print(x);puts("this is y");print(y);
            printf("%lld\n",(ll)(size[y]-size[x])*size[x]);
        }
    }//for (int i=1;i<=n;++i) printf("%d ",size[i]);
    return 0;
}