[ZJOI2012][bzoj 2816] 网络 network [LCT]

题目：

http://www.lydsy.com/JudgeOnline/problem.php?id=2816

思路：

第一个条件看完暂时还没什么想法

看完第二个，发现每一个颜色都是一个森林

进而想到对于每一个颜色维护LCT

看数据范围，n<=10000, c<=10，可行

对于操作0，把每一个LCT上该店的权值都修改

对于操作1，先检测这条边是否存在，若不存在就No such edge

如果这条边存在，且原来的颜色不同于要修改的颜色的话，就先判断Error 1和Error 2

如果要改的边已经是目标颜色了，就特判跳过

上述判断都通过了就cut和link即可

对于操作2，每一个节点维护一个最大值，输出即可

Code：

/**************************************************************
    Problem: 2816
    User: dedicatus545
    Language: C++
    Result: Accepted
    Time:8032 ms
    Memory:6688 kb
****************************************************************/
 
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<unistd.h>
#define col(i,c) n*c+i
using namespace std;
inline int read(){
    int re=0,flag=1;char ch=getchar();
    while(ch>'9'||ch<'0'){
        if(ch=='-') flag=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9') re=(re<<1)+(re<<3)+ch-'0',ch=getchar();
    return re*flag;
}
struct edge{
    int to,next,w;
}e[200010];
int n,m,C,cnt=-1,q,first[100010],color[10010][11];
int fa[100010]={0},ch[100010][2]={0},w[100010],a[100010];
bool rev[100010]={0},rt[100010]={0};
void _swap(int &l,int &r){l^=r;r^=l;l^=r;}
void update(int x){a[x]=max(a[ch[x][0]],max(a[ch[x][1]],w[x]));};
void pushrev(int x){
    if(!x) return;
    _swap(ch[x][0],ch[x][1]);
    rev[x]^=1;
}
void pushdown(int x){
    if(!x) return;
    if(rev[x]){
        pushrev(ch[x][0]);
        pushrev(ch[x][1]);
        rev[x]=0;
    }
}
void push(int x){
    //cout<<"push "<<x<<' '<<rt[x]<<'
';
    if(!x) sleep(1000);
    if(!rt[x]) push(fa[x]);
    pushdown(x);
}
int get(int x){return ch[fa[x]][1]==x;}
void rotate(int x){
    //cout<<"rotate "<<x<<'
';
    int f=fa[x],ff=fa[f],son=get(x);
    ch[f][son]=ch[x][son^1];
    if(ch[f][son]) fa[ch[f][son]]=f;
    ch[x][son^1]=f;fa[f]=x;
    fa[x]=ff;
    if(rt[f]) rt[x]=1,rt[f]=0;
    else ch[ff][ch[ff][1]==f]=x;
    update(f);update(x);
}
void splay(int x){
    //cout<<"splay "<<x<<'
';
    push(x);
    for(int f;!rt[x];rotate(x))
        if(!rt[f=fa[x]])
            rotate((get(x)==get(f))?f:x);
    update(x);
}
void access(int x){
    //cout<<"access "<<x<<'
';
    int y=0;
    do{
        splay(x);
        rt[ch[x][1]]=1;
        rt[ch[x][1]=y]=0;
        x=fa[y=x];
        update(x);
    }while(x);
}
void makeroot(int x){
    //cout<<"makeroot "<<x<<'
';
    access(x);splay(x);pushrev(x);
}
bool judge(int x,int y){
    while(fa[x]) x=fa[x];
    while(fa[y]) y=fa[y];
    return x==y;
}
void init(int x,int y){
    //cout<<"init "<<x<<' '<<y<<'
';
    makeroot(x);fa[x]=y;
}
int link(int x,int y){
    if(judge(x,y)) return 0; 
    makeroot(x);fa[x]=y;
    return 1;
}
int cut(int x,int y){
    if(!judge(x,y)) return 0;
    makeroot(x);splay(y);
    fa[ch[y][0]]=fa[y];
    rt[ch[y][0]]=1;
    fa[y]=0;ch[y][0]=0;
    return 1;
}
void add(int u,int v,int w){
    e[++cnt]=(edge){v,first[u],w};first[u]=cnt;
    e[++cnt]=(edge){u,first[v],w};first[v]=cnt;
}
int split(int u,int v){
    if(!judge(u,v)) return -1;
    makeroot(u);access(v);splay(v);
    return a[v];
}
int main(){
   // freopen("networkzj.in","r",stdin);
  //  freopen("networkzj.out","w",stdout);
    memset(first,-1,sizeof(first));
    int i,j,t1,t2,t3,t4;
    n=read();m=read();C=read();q=read();
    for(i=1;i<=n;i++){
        t1=read();
        for(j=0;j<C;j++) a[col(i,j)]=w[col(i,j)]=t1,rt[col(i,j)]=1;
    }
    
    for(i=1;i<=m;i++){
        t1=read();t2=read();t3=read();
        init(col(t1,t3),col(t2,t3));
        add(t1,t2,t3);
        color[t1][t3]++;color[t2][t3]++;
    }
    //for(i=1;i<=n;i++){
        //for(j=0;j<C;j++) cout<<color[i][j]<<' ';
        //cout<<'
';
    //}
    for(i=1;i<=q;i++){
        //cout<<"operation "<<i<<'
';
        t1=read();
        if(t1==0){
            t2=read();t3=read();
            for(j=0;j<C;j++){
                makeroot(col(t2,j));w[col(t2,j)]=t3;update(col(t2,j));
            }
        }
        if(t1==1){
            t2=read();t3=read();t4=read();bool flag=1,f=1;
            for(j=first[t2];~j;j=e[j].next) 
                if(e[j].to==t3){
                    flag=0;
                    if(e[j].w==t4) f=0;
                }
            if(flag){
                printf("No such edge.
");
                continue;
            } 
            if(f&&(color[t2][t4]==2||color[t3][t4]==2)){
                printf("Error 1.
");continue;
            }
            if(f&&(judge(col(t2,t4),col(t3,t4)))){
                printf("Error 2.
");continue;
            }
            for(j=first[t2];~j;j=e[j].next){
                if(e[j].to==t3){
                    color[t2][e[j].w]--;color[t3][e[j].w]--;
                    cut(col(t2,e[j].w),col(t3,e[j].w));
                    e[j].w=e[j^1].w=t4;
                    color[t2][t4]++;color[t3][t4]++;
                    link(col(t2,t4),col(t3,t4));
                    printf("Success.
");break;
                } 
            }
        }
        if(t1==2){
            t2=read();t3=read();t4=read();
            printf("%d
",split(col(t3,t2),col(t4,t2)));
        }
    }
}