码迷,mamicode.com
首页 > 其他好文 > 详细

[BJOI2014] 大融合 - Link Cut Tree

时间:2019-10-19 21:03:57      阅读:94      评论:0      收藏:0      [点我收藏+]

标签:维护   maker   access   lse   cut   ons   hup   就是   pre   

经典也是最基础的Link Cut Tree维护子树问题。我们考虑维护子树有哪些不同,无非就是需要去维护虚儿子,在这里就是维护虚儿子的size,记为si

  • pushup(x) 这时的更新来源除了左右儿子和自己以外还有虚儿子
  • access(x) 由于access操作是做虚实互换的过程,会对虚儿子数据产生影响
  • link(x,y) 授予了新的父子关系,但这种关系是通过虚儿子实现的

需要注意 cut(x,y) 本身虽然没有影响虚儿子但也要更新一下

#include <bits/stdc++.h>
using namespace std;

const int N = 100005;

int n,m,val[N];

struct LinkCutTree {
    int top, q[N], ch[N][2], fa[N], xr[N], rev[N], s[N], si[N];
    inline void pushup(int x) { // !
        s[x]=s[ch[x][0]]+s[ch[x][1]]+si[x]+1;
    }
    inline void pushdown(int x) {
        if(!rev[x])
            return;
        rev[ch[x][0]]^=1;
        rev[ch[x][1]]^=1;
        rev[x]^=1;
        swap(ch[x][0],ch[x][1]);
    }
    inline bool isroot(int x) {
        return ch[fa[x]][0]!=x && ch[fa[x]][1]!=x;
    }
    inline void rotate(int p) {
        int q=fa[p], y=fa[q], x=ch[fa[p]][1]==p;
        ch[q][x]=ch[p][x^1];
        fa[ch[q][x]]=q;
        ch[p][x^1]=q;
        fa[q]=p;
        fa[p]=y;
        if(y)
            if(ch[y][0]==q)
                ch[y][0]=p;
            else  if(ch[y][1]==q)
                ch[y][1]=p;
        pushup(q);
        pushup(p);
    }
    inline void splay(int x) {
        q[top=1]=x;
        for(int i=x; !isroot(i); i=fa[i])
            q[++top]=fa[i];
        for(int i=top; i; i--)
            pushdown(q[i]);
        for(; !isroot(x); rotate(x))
            if(!isroot(fa[x]))
                rotate((ch[fa[x]][0]==x)==(ch[fa[fa[x]]][0]==fa[x])?fa[x]:x);
    }
    void access(int x) { // !
        for(int t=0; x; t=x,x=fa[x])
            splay(x),si[x]+=s[ch[x][1]],si[x]-=s[ch[x][1]=t],pushup(x);
    }
    void makeroot(int x) {
        access(x);
        splay(x);
        rev[x]^=1;
    }
    int find(int x) {
        access(x);
        splay(x);
        while(ch[x][0])
            x=ch[x][0];
        return x;
    }
    void split(int x,int y) {
        makeroot(x);
        access(y);
        splay(y);
    }
    void cut(int x,int y) { // !
        split(x,y);
        if(ch[y][0]==x)
            ch[y][0]=0, fa[x]=0;
        pushup(y);
    }
    void link(int x,int y) { // !
        split(x,y);
        si[fa[x]=y]+=s[x];
        pushup(y);
    }
} T;

int main() {
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1; i<=n; i++)
        T.s[i]=1;
    for(int i=1; i<=m; i++) {
        string str;
        cin>>str;
        int u,v;
        cin>>u>>v;
        if(str[0]=='A') {
            T.link(u,v);
        } else {
            T.cut(u,v);
            T.makeroot(u);
            T.makeroot(v);
            cout<<T.s[u]*T.s[v]<<endl;
            T.link(u,v);
        }
    }
}

[BJOI2014] 大融合 - Link Cut Tree

标签:维护   maker   access   lse   cut   ons   hup   就是   pre   

原文地址:https://www.cnblogs.com/mollnn/p/11704915.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!