BZOJ 4530: [Bjoi2014]大融合(lct)

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

解题思路

  \(lct\)维护子树信息裸题。\(lct\)维护子树信息时需要记录一下虚子树的信息和虚子树+实子树+自己的信息。虚子树信息只有在\(link\)\(access\)的时候才会被改变。具体看代码。

代码

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

using namespace std;
const int MAXN = 100005;
typedef long long LL;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
}

int n,q,head[MAXN],cnt,to[MAXN<<1],nxt[MAXN<<1];
int siz[MAXN],Siz[MAXN],ch[MAXN][2],fa[MAXN];
bool rev[MAXN];

inline void pushup(int x){
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1+Siz[x];
}

inline void pushdown(int x){
    if(rev[x]){
        swap(ch[x][0],ch[x][1]);
        if(ch[x][0]) rev[ch[x][0]]^=1;
        if(ch[x][1]) rev[ch[x][1]]^=1;
        rev[x]=0;
    }
}

inline bool isroot(int x){
    return (x!=ch[fa[x]][0] && x!=ch[fa[x]][1]);
}

void pd(int x){
    if(fa[x]) pd(fa[x]);pushdown(x);
}

inline bool check(int x){
    return (x==ch[fa[x]][1]);
}

inline void rotate(int x){
    int y=fa[x],z=fa[y];bool chk=check(x);
    if(!isroot(y)) ch[z][check(y)]=x;
    ch[y][chk]=ch[x][chk^1];fa[ch[x][chk^1]]=y;
    ch[x][chk^1]=y;fa[y]=x;fa[x]=z;pushup(y);pushup(x);
}

inline void splay(int x){
    pd(x);
    for(;!isroot(x);rotate(x))
        if(!isroot(fa[x])) rotate(check(fa[x])==check(x)?fa[x]:x);
}

inline void access(int x){
    for(int y=0;x;y=x,x=fa[x]){
        splay(x);if(ch[x][1]) Siz[x]+=siz[ch[x][1]];
        if(y) Siz[x]-=siz[y];ch[x][1]=y;pushup(x);
    }
}

inline void makeroot(int x){
    access(x);splay(x);rev[x]^=1;
}

inline void link(int x,int y){
    makeroot(x);makeroot(y);Siz[y]+=siz[x];fa[x]=y;pushup(y);
}

int main(){
    n=rd(),q=rd();char op;int x,y;
    for(int i=1;i<=n;i++) siz[i]=1;
    while(q--){
        cin>>op;x=rd(),y=rd();
        if(op=='A') link(x,y);
        if(op=='Q') {
            makeroot(x);access(y);splay(x);
            printf("%lld\n",(LL)(Siz[y]+1)*(siz[x]-Siz[y]-1));
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/sdfzsyq/p/10032869.html