ZJOI2012 网络——LCT相关题目

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

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

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

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

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

2. 修改一条边的颜色。

3. 查询由颜色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[],mark,max,num;
 };
 struct LCT{
     dt data[];
     int tot;
     void Init(){
         data[].fa=data[].mark=data[].max=data[].max=data[].num=;
         tot=;
     }
     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[]=;
         up(y);
     }
     void access(int x){
         int y=;
         while(x){
             splay(x);
             data[x].ch[]=y;
             up(x);
             y=x;x=data[x].fa;
         }
     }
     void make_root(int x){
         access(x);splay(x);data[x].mark^=;
     }
     int find_root(int x){
         access(x);splay(x);
         while(data[x].ch[])
             x=data[x].ch[];
         return x;
     }
     void splay(int x){
         push(x);
         int fa,fafa;
         for(fa=data[x].fa;data[fa].ch[]==x||data[fa].ch[]==x;roll(x),fa=data[x].fa){
             fafa=data[fa].fa;
             if(data[fafa].ch[]==fa||data[fafa].ch[]==fa)
                 if((data[fa].ch[]==x)^(data[fafa].ch[]==fa))
                     roll(x);
                 else
                     roll(fa);
         }
     }
     void roll(int now){
         int fa=data[now].fa,fafa=data[fa].fa,wh=data[fa].ch[]==now;
         data[fa].ch[wh]=data[now].ch[wh^];data[data[fa].ch[wh]].fa=fa;
         data[now].ch[wh^]=fa;data[fa].fa=now;
         data[now].fa=fafa;
         if (data[fafa].ch[]==fa||data[fafa].ch[]==fa)
             data[fafa].ch[data[fafa].ch[]==fa]=now;
         up(fa);up(now);
     }
     void push(int x){
         int fa=data[x].fa;
         if(data[fa].ch[]==x||data[fa].ch[]==x)
             push(fa);
         down(x);
     }
     void up(int x){
         int a=data[data[x].ch[]].max>data[data[x].ch[]].max?data[data[x].ch[]].max:data[data[x].ch[]].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[],data[x].ch[]=data[x].ch[],data[x].ch[]=i;
             data[x].mark=;
             data[data[x].ch[]].mark^=;data[data[x].ch[]].mark^=;
         }
     }
 };
 int Hash(int ,int ,int );
 LCT lct[];
 int col[][];
 int hash[];
 int poll[],next[],colway[],tot;
 int main()
 {
     int i,j,k,u,v,w;
     scanf("%d%d%d%d",&n,&m,&c,&p);
     for(j=;j<c;j++)lct[j].Init();
     for(i=;i<=n;i++){
         scanf("%d",&k);
         for(j=;j<c;j++)
             lct[j].data[i].num=lct[j].data[i].max=k;
     }
     for(i=;i<=m;i++){
         scanf("%d%d%d",&u,&v,&w);
         lct[w].link(u,v);
         col[u][w]++;col[v][w]++;
         Hash(u*+v,,w),Hash(v*+u,,w);
     }
     for(i=;i<=p;i++){
         scanf("%d",&j);
         if(j==){
             scanf("%d%d",&u,&v);
             for(k=;k<c;k++){
                 lct[k].splay(u);
                 lct[k].data[u].num=v;
                 lct[k].up(u);
             }
             continue;
         }
         if(j==){
             int lk;
             scanf("%d%d%d",&u,&v,&k);
             lk=Hash(u*+v,,-);
             if(!lk){
                 printf("No such edge.\n");continue;
             }
             if(--lk==k){
                 printf("Success.\n");continue;
             }
             if(col[u][k]==||col[v][k]==){
                 printf("Error 1.\n");continue;
             }
             lct[k].make_root(u);
             if(lct[k].find_root(v)==u){
                 printf("Error 2.\n");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*+v,,k);Hash(v*+u,,k);
             printf("Success.\n");continue;
         }
         if(j==){
             scanf("%d%d%d",&k,&u,&v);
             lct[k].make_root(u);
             if(lct[k].find_root(v)!=u){
                 printf("-1\n");continue;
             }
             lct[k].access(v);
             lct[k].splay(v);
             printf("%d\n",lct[k].data[v].max);
             continue;
         }
     }
     return ;
 }
 int Hash(int sum,int p,int color){
     int mod=,x=sum%mod,i;
     if(!hash[x]){
         if(!p)return ;
         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 ;
         }
         if(poll[i]!=sum)return ;
         if(color==-)return colway[i]+;
         colway[i]=color;
     }
     return ;
 }