ZJOI2012 网络——LCT相关题目

有一个无向图G，每个点有个权值，每条边有一个颜色。这个无向图满足以下两个条件：

1. 对于任意节点连出去的边中，相同颜色的边不超过两条。

2. 图中不存在同色的环，同色的环指相同颜色的边构成的环。

在这个图上，你要支持以下三种操作：

0. 修改一个节点的权值。

1. 修改一条边的颜色。

2. 查询由颜色c的边构成的图中，所有可能在节点u到节点v之间的简单路径上的节点的权值的最大值。

https://daniu.luogu.org/problem/show?pid=2173

                                                                                 -by luogu

---

---

对每个颜色建LCT，对error1数组记录，对error2LCT判联通性，对NO Such...直接写了hash，trie树大概开不下；

发现封装起来非常好；

代码:

#include<cstdio>
#include<cstring>
using namespace std;
int n,m,c,p;
struct dt{
    int fa,ch[2],mark,max,num;
};
struct LCT{
    dt data[10010];
    int tot;
    void Init(){
        data[0].fa=data[0].mark=data[0].max=data[0].max=data[0].num=0;
        tot=0;
    }
    void link(int x,int y){
        make_root(x);data[x].fa=y;
    }
    void cut(int x,int y){
        make_root(x),access(y),splay(y);
        data[x].fa=data[y].ch[0]=0;
        up(y);
    }
    void access(int x){
        int y=0;
        while(x){
            splay(x);
            data[x].ch[1]=y;
            up(x);
            y=x;x=data[x].fa;
        }
    }
    void make_root(int x){
        access(x);splay(x);data[x].mark^=1;
    }
    int find_root(int x){
        access(x);splay(x);
        while(data[x].ch[0])
            x=data[x].ch[0];
        return x;
    }
    void splay(int x){
        push(x);
        int fa,fafa;
        for(fa=data[x].fa;data[fa].ch[1]==x||data[fa].ch[0]==x;roll(x),fa=data[x].fa){
            fafa=data[fa].fa;
            if(data[fafa].ch[1]==fa||data[fafa].ch[0]==fa)
                if((data[fa].ch[0]==x)^(data[fafa].ch[0]==fa))
                    roll(x);
                else
                    roll(fa);
        }
    }
    void roll(int now){
        int fa=data[now].fa,fafa=data[fa].fa,wh=data[fa].ch[1]==now;
        data[fa].ch[wh]=data[now].ch[wh^1];data[data[fa].ch[wh]].fa=fa;
        data[now].ch[wh^1]=fa;data[fa].fa=now;
        data[now].fa=fafa;
        if (data[fafa].ch[0]==fa||data[fafa].ch[1]==fa)
            data[fafa].ch[data[fafa].ch[1]==fa]=now;
        up(fa);up(now);
    }
    void push(int x){
        int fa=data[x].fa;
        if(data[fa].ch[0]==x||data[fa].ch[1]==x)
            push(fa);
        down(x);
    }
    void up(int x){
        int a=data[data[x].ch[0]].max>data[data[x].ch[1]].max?data[data[x].ch[0]].max:data[data[x].ch[1]].max;
        data[x].max=data[x].num>a?data[x].num:a;
    }
    void down(int x){
        int i;
        if(data[x].mark){
            i=data[x].ch[0],data[x].ch[0]=data[x].ch[1],data[x].ch[1]=i;
            data[x].mark=0;
            data[data[x].ch[0]].mark^=1;data[data[x].ch[1]].mark^=1;
        }
    }
};
int Hash(int ,int ,int );
LCT lct[10];
int col[10010][10];
int hash[100000];
int poll[200010],next[200010],colway[200010],tot;
int main()
{
    int i,j,k,u,v,w;
    scanf("%d%d%d%d",&n,&m,&c,&p);
    for(j=0;j<c;j++)lct[j].Init();
    for(i=1;i<=n;i++){
        scanf("%d",&k);
        for(j=0;j<c;j++)
            lct[j].data[i].num=lct[j].data[i].max=k;
    }
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&u,&v,&w);
        lct[w].link(u,v);
        col[u][w]++;col[v][w]++;
        Hash(u*100001+v,1,w),Hash(v*100001+u,1,w);
    }
    for(i=1;i<=p;i++){
        scanf("%d",&j);
        if(j==0){
            scanf("%d%d",&u,&v);
            for(k=0;k<c;k++){
                lct[k].splay(u);
                lct[k].data[u].num=v;
                lct[k].up(u);
            }
            continue;
        }
        if(j==1){
            int lk;
            scanf("%d%d%d",&u,&v,&k);
            lk=Hash(u*100001+v,0,-1);
            if(!lk){
                printf("No such edge.
");continue;
            }
            if(--lk==k){
                printf("Success.
");continue;
            }
            if(col[u][k]==2||col[v][k]==2){
                printf("Error 1.
");continue;
            }
            lct[k].make_root(u);
            if(lct[k].find_root(v)==u){
                printf("Error 2.
");continue;
            }
            lct[lk].cut(u,v);
            lct[k].link(u,v);
            col[u][lk]--;col[v][lk]--;col[u][k]++;col[v][k]++;
            Hash(u*100001+v,0,k);Hash(v*100001+u,0,k);
            printf("Success.
");continue;
        }
        if(j==2){
            scanf("%d%d%d",&k,&u,&v);
            lct[k].make_root(u);
            if(lct[k].find_root(v)!=u){
                printf("-1
");continue;
            }
            lct[k].access(v);
            lct[k].splay(v);
            printf("%d
",lct[k].data[v].max);
            continue;
        }
    }
    return 0;
}
int Hash(int sum,int p,int color){
    int mod=99991,x=sum%mod,i;
    if(!hash[x]){
        if(!p)return 0;
        hash[x]=++tot;
        poll[tot]=sum;colway[tot]=color;
    }
    else{
        i=hash[x];
        while(next[i]&&poll[i]!=sum)
            i=next[i];
        if(p){
            next[i]=++tot;
            poll[tot]=sum;
            colway[tot]=color;
            return 0;
        }
        if(poll[i]!=sum)return 0;
        if(color==-1)return colway[i]+1;
        colway[i]=color;
    }
    return 0;
}