bzoj 2816 [ZJOI2012]网络 - LCT

持续屯板子……

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<utility>
#define N 10010
#define C 12
#define mp make_pair
using namespace std;
typedef pair<int,int> pii;
int d[N][C];
struct LCT{
    int val[N],maxv[N],rev[N],fa[N],pf[N],ch[N][2];
    inline int gw(int x)
    {
        return ch[fa[x]][1]==x;
    }
    inline int push_down(int x)
    {
        if(!rev[x]) return 0;
        swap(ch[x][0],ch[x][1]);
        if(ch[x][0]) rev[ch[x][0]]^=1;
        if(ch[x][1]) rev[ch[x][1]]^=1;
        return rev[x]=0;
    }
    int all_down(int x)
    {
        if(fa[x]) all_down(fa[x]);
        if(rev[x]) push_down(x);
        return 0;
    }
    inline int push_up(int x)
    {
        return maxv[x]=max(max(maxv[ch[x][0]],maxv[ch[x][1]]),val[x]);
    }
    inline int setc(int x,int y,int z)
    {
        if(!x) return fa[y]=0;
        ch[x][z]=y;if(y) fa[y]=x;
        return push_up(x);
    }
    inline int rotate(int x)
    {
        int y=fa[x],z=fa[y],a=gw(x),b=gw(y),c=ch[x][a^1];
        return swap(pf[x],pf[y]),setc(y,c,a),setc(x,y,a^1),setc(z,x,b);
    }
    inline int splay(int x)
    {
        all_down(x);
        while(fa[x])
        {
            if(fa[fa[x]])
            {
                if(gw(x)^gw(fa[x])) rotate(x);
                else rotate(fa[x]);
            }
            rotate(x);
        }
        return 0;
    }
    inline int expose(int x)
    {
        splay(x);int c=ch[x][1];
        if(c) fa[c]=ch[x][1]=0,pf[c]=x;
        return push_up(x);
    }
    inline int splice(int x)
    {
        splay(x);int y=pf[x];if(!y) return false;
        expose(y),splay(y),setc(y,x,1),pf[x]=0;
        return true;
    }
    inline int access(int x)
    {
        expose(x);while(splice(x));return 0;
    }
    inline int evert(int x)
    {
        return access(x),splay(x),rev[x]^=1;
    }
    inline int get_r(int x)
    {
        access(x),splay(x);
        while(ch[x][0]) x=ch[x][0];
        return splay(x),x;
    }
    inline int connected(int x,int y)
    {
        return get_r(x)==get_r(y);
    }
    inline int link(int x,int y)
    {
        return evert(y),splay(y),pf[y]=x;
    }
    inline int cut(int x,int y)
    {
        evert(x),access(y),splay(x);
        ch[x][1]=fa[y]=pf[y]=0;
        return push_up(x);
    }
    inline int update(int x,int v)
    {
        return splay(x),val[x]=v,push_up(x);
    }
    inline int query(int x,int y)
    {
        return evert(x),access(y),splay(x),maxv[x];
    }
}t[C];
map<pii,int> es;
#define gc getchar()
inline int inn()
{
    int x,ch;while((ch=gc)<'0'||ch>'9');
    x=ch^'0';while((ch=gc)>='0'&&ch<='9')
        x=(x<<1)+(x<<3)+(ch^'0');return x;
}
int main()
{
    int n=inn(),m=inn(),c=inn(),k=inn();
    for(int i=1;i<=n;i++)
        for(int j=0,v=inn();j<c;j++)
            t[j].maxv[i]=t[j].val[i]=v;
    for(int i=1,u,v,w;i<=m;i++)
        u=inn(),v=inn(),w=inn(),t[w].link(u,v),d[u][w]++,d[v][w]++,
        es.insert(mp(mp(u,v),w)),es.insert(mp(mp(v,u),w));
    while(k--)
    {
        int opt=inn();
        if(!opt) for(int i=0,x=inn(),y=inn();i<c;i++) t[i].update(x,y);
        else if(opt==1)
        {
            int x=inn(),y=inn(),w=inn(),lw=0;
            if(!es.count(mp(x,y))) { printf("No such edge.\n");continue; }
            if(w==(lw=es[mp(x,y)])) { printf("Success.\n");continue; }
            if(d[x][w]==2||d[y][w]==2) { printf("Error 1.\n");continue; }
            if(t[w].connected(x,y)) { printf("Error 2.\n");continue; }
            d[x][w]++,d[y][w]++,d[x][lw]--,d[y][lw]--,printf("Success.\n");
            es[mp(x,y)]=es[mp(y,x)]=w,t[lw].cut(x,y),t[w].link(x,y);
        }
        else{
            int w=inn(),x=inn(),y=inn();
            if(!t[w].connected(x,y)) printf("-1\n");
            else printf("%d\n",t[w].query(x,y));
        }
    }
    return 0;
}