BZOJ2816: [ZJOI2012]网络（洛谷P2173）

LCT

BZOJ题目传送门
洛谷题目传送门

这。。。就是裸题啊

既然$c\leq10$，那么我们建 c c <script type="math/tex" id="MathJax-Element-2">c</script>颗LCT就好了。

对于刚开始的边，按照边的颜色把两个节点link起来就好了。

对于操作0：直接改就好了。。。
对于操作1：找到这条边在哪个LCT上，然后分情况讨论就好了，具体看代码。
对于操作2：直接查就好了。。。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define M 10
#define F inline
#define V F void
using namespace std;
struct tree{ int to[2],mx,fa,f; }t[M][N];
int n,m,k,c,w,stk[N],tp,a[N],dg[M][N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
V writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
V _write(int x){ writec(x),putchar('\n'); }
#define rt(x) (t[c][t[c][x].fa].to[0]!=(x)&&t[c][t[c][x].fa].to[1]!=(x))
V pshp(int x){
    t[c][x].mx=max(a[x],max(t[c][t[c][x].to[0]].mx,t[c][t[c][x].to[1]].mx));
}
V pshd(int x){
    if (!t[c][x].f) return; t[c][x].f=0;
    int &l=t[c][x].to[0],&r=t[c][x].to[1];
    t[c][l].f^=1,t[c][r].f^=1,swap(l,r);
}
V rtt(int x){
    int y=t[c][x].fa,z=t[c][y].fa,l=(t[c][y].to[0]==x);
    if (!rt(y)) t[c][z].to[t[c][z].to[0]!=y]=x;
    t[c][x].fa=z,t[c][y].fa=x,t[c][t[c][x].to[l]].fa=y;
    t[c][y].to[l^1]=t[c][x].to[l],t[c][x].to[l]=y;
    pshp(y),pshp(x);
}
V splay(int x){
    stk[tp=1]=x;
    for (int i=x;!rt(i);i=t[c][i].fa) stk[++tp]=t[c][i].fa;
    while (tp) pshd(stk[tp--]);
    for (int y=t[c][x].fa,z=t[c][y].fa;!rt(x);rtt(x),y=t[c][x].fa,z=t[c][y].fa)
        if (!rt(y)) rtt((t[c][z].to[0]==y^t[c][y].to[0]==x)?x:y);
}
V ccss(int x){ for (int i=0;x;i=x,x=t[c][x].fa) splay(x),t[c][x].to[1]=i,pshp(x); }
V mkrt(int x){ ccss(x),splay(x),t[c][x].f^=1; }
V lnk(int x,int y){ dg[c][x]++,dg[c][y]++,mkrt(x),t[c][x].fa=y; }
V sprt(int x,int y){ mkrt(x),ccss(y),splay(y); }
V cut(int x,int y){ dg[c][x]--,dg[c][y]--,sprt(x,y),t[c][y].to[0]=t[c][x].fa=0; }
F int find(int x){ ccss(x),splay(x); while (t[c][x].to[0]) x=t[c][x].to[0]; return x; }
int main(){
    n=_read(),m=_read(),w=_read(),k=_read();
    for (int i=1;i<=n;i++) a[i]=_read();
    for (int i=1,x,y;i<=m;i++)
        x=_read(),y=_read(),c=_read(),lnk(x,y);
    for (int f,x,y,p;k;k--)
        switch (f=_read()){
            case 0:
                x=_read(),p=_read();
                for (c=0;c<w;c++) ccss(x),splay(x);
                a[x]=p; for (c=0;c<w;c++) pshp(x); break;
            case 1:
                x=_read(),y=_read(),p=_read();
                for (c=0;c<w;c++)
                    if (find(x)==find(y)){
                        sprt(x,y); int i=c;
                        if (t[c][y].to[0]!=x||t[c][x].to[1]) continue;
                        if (c==p) { puts("Success."); break; }
                        if (dg[p][x]==2||dg[p][y]==2) { puts("Error 1."); break; }
                        c=p; if (find(x)==find(y)) { puts("Error 2."); break; }
                        c=i,cut(x,y),c=p,lnk(x,y),puts("Success."),c=i; break;
                    }
                if (c==w) puts("No such edge."); break;
            case 2:
                c=_read(),x=_read(),y=_read();
                if (find(x)!=find(y)){ puts("-1"); break; }
                sprt(x,y);_write(t[c][y].mx); break;
        }
    return 0;
}