[BZOJ2816][ZJOI2012]网络（lct+map）

题目描述

题目描述 Description

有一个无向图G，每个点有个权值，每条边有一个颜色。这个无向图满足以下两个条件：
1. 对于任意节点连出去的边中，相同颜色的边不超过两条。
2. 图中不存在同色的环，同色的环指相同颜色的边构成的环。
在这个图上，你要支持以下三种操作：
0. 修改一个节点的权值。
1. 修改一条边的颜色。
2. 查询由颜色c的边构成的图中，所有可能在节点u到节点v之间的简单路径上的节点的权值的最大值。

输入描述 Input Description

第一行包含四个正整数N, M, C, K，其中N为节点个数，M为边数，C为边的颜色数，K为操作数。
接下来N行，每行一个正整数vi，为节点i的权值。
之后M行，每行三个正整数u, v, w，为一条连接节点u和节点v的边，颜色为w。满足1 ≤ u, v ≤ N，0 ≤ w < C，保证u ≠ v，且任意两个节点之间最多存在一条边(无论颜色)。
最后K行，每行表示一个操作。每行的第一个整数k表示操作类型。
0. k = 0为修改节点权值操作，之后两个正整数x和y，表示将节点x的权值vx修改为y。
1. k = 1为修改边的颜色操作，之后三个正整数u, v和w，表示将连接节点u和节点v的边的颜色修改为颜色w。满足0 ≤ w < C。
2. k = 2为查询操作，之后三个正整数c, u和v，表示查询所有可能在节点u到节点v之间的由颜色c构成的简单路径上的节点的权值的最大值。如果不存在u和v之间不存在由颜色c构成的路径，那么输出“-1”。

输出描述 Output Description

包含若干行，每行输出一个对应的信息。
1. 对于修改节点权值操作，不需要输出信息。
2. 对于修改边的颜色操作，按以下几类输出：
a) 若不存在连接节点u和节点v的边，输出“No such edge.”。
b) 若修改后不满足条件1，不修改边的颜色，并输出“Error 1.”。
c) 若修改后不满足条件2，不修改边的颜色，并输出“Error 2.”。
d) 其他情况，成功修改边的颜色，并输出“Success.”。
输出满足条件的第一条信息即可，即若同时满足b和c，则只需要输出“Error 1.”。
3. 对于查询操作，直接输出一个整数。

样例输入 Sample Input

4 5 2 7
1
2
3
4
1 2 0
1 3 1
2 3 0
2 4 1
3 4 0
2 0 1 4
1 1 2 1
1 4 3 1
2 0 1 4
1 2 3 1
0 2 5
2 1 1 4

样例输出 Sample Output

4
Success.
Error 2.
-1
Error 1.
5

数据范围及提示 Data Size & Hint

对于30%的数据：N ≤ 1000，M ≤ 10000，C ≤ 10，K ≤ 1000。
另有20%的数据：N ≤ 10000，M ≤ 100000，C = 1，K ≤ 100000。
对于100%的数据：N ≤ 10000，M ≤ 100000，C ≤ 10，K ≤ 100000。

题解

lct裸题
维护c棵lct然后注意判断一下各种条件就行了
我判断是否有边的方式是用map，然后记录一下每一个点在每一种颜色的图中的度

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
using namespace std;
#define N 10005

struct data
{
    int x,y;
    bool operator < (const data &a) const
    {
        return x<a.x||(x==a.x&&y<a.y);
    }
};
map <data,int> ext,col;
int n,m,c,k;
int f[15][N],ch[15][N][2],rev[15][N],maxn[15][N],val[15][N],du[15][N],stack[N];

bool isroot(int cc,int x)
{
    return ch[cc][f[cc][x]][0]!=x&&ch[cc][f[cc][x]][1]!=x;
}
int get(int cc,int x)
{
    return ch[cc][f[cc][x]][1]==x;
}
void update(int cc,int x)
{
    maxn[cc][x]=val[cc][x];
    if (ch[cc][x][0]&&maxn[cc][ch[cc][x][0]]>maxn[cc][x]) maxn[cc][x]=maxn[cc][ch[cc][x][0]];
    if (ch[cc][x][1]&&maxn[cc][ch[cc][x][1]]>maxn[cc][x]) maxn[cc][x]=maxn[cc][ch[cc][x][1]];
}
void pushdown(int cc,int x)
{
    if (x&&rev[cc][x])
    {
        if (ch[cc][x][0]) rev[cc][ch[cc][x][0]]^=1;
        if (ch[cc][x][1]) rev[cc][ch[cc][x][1]]^=1;
        swap(ch[cc][x][0],ch[cc][x][1]);
        rev[cc][x]=0;
    }
}
void rotate(int cc,int x)
{
    int old=f[cc][x],oldf=f[cc][old],wh=get(cc,x);
    if (!isroot(cc,old)) ch[cc][oldf][ch[cc][oldf][1]==old]=x;
    f[cc][x]=oldf;
    ch[cc][old][wh]=ch[cc][x][wh^1];
    if (ch[cc][old][wh]) f[cc][ch[cc][old][wh]]=old;
    ch[cc][x][wh^1]=old;
    f[cc][old]=x;
    update(cc,old);
    update(cc,x);
}
void splay(int cc,int x)
{
    int top=0;stack[++top]=x;
    for (int i=x;!isroot(cc,i);i=f[cc][i]) stack[++top]=f[cc][i];
    for (int i=top;i;--i) pushdown(cc,stack[i]);

    for (int fa;!isroot(cc,x);rotate(cc,x))
        if (!isroot(cc,fa=f[cc][x]))
            rotate(cc,(get(cc,x)==get(cc,fa))?fa:x);
}
void access(int cc,int x)
{
    int t=0;
    for (;x;t=x,x=f[cc][x])
    {
        splay(cc,x);
        ch[cc][x][1]=t;
        update(cc,x);
    }
}
void reverse(int cc,int x)
{
    access(cc,x);
    splay(cc,x);
    rev[cc][x]^=1;
}
int find(int cc,int x)
{
    access(cc,x);
    splay(cc,x);
    while (ch[cc][x][0]) x=ch[cc][x][0];
    return x;
}
void link(int cc,int x,int y)
{
    reverse(cc,x);
    f[cc][x]=y;
}
void cut(int cc,int x,int y)
{
    reverse(cc,x);
    access(cc,y);
    splay(cc,y);
    ch[cc][y][0]=f[cc][x]=0;
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&c,&k);
    for (int i=1;i<=n;++i)
    {
        int x;scanf("%d",&x);
        for (int j=0;j<c;++j) val[j][i]=x;
    }
    for (int i=1;i<=m;++i)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        if (x>y) swap(x,y);
        link(z,x,y);
        data now=(data){x,y};
        col[now]=z;ext[now]=1;
        ++du[z][x];++du[z][y];
    }
    for (int i=1;i<=k;++i)
    {
        int opt;scanf("%d",&opt);
        if (opt==0)
        {
            int x,y;scanf("%d%d",&x,&y);
            for (int j=0;j<c;++j)
            {
                val[j][x]=y;
                splay(j,x);
            }
        }
        else if (opt==1)
        {
            int x,y,z;scanf("%d%d%d",&x,&y,&z);
            if (x>y) swap(x,y);
            data now=(data){x,y};
            if (!ext[now])
            {
                puts("No such edge.");
                continue;
            }
            int cc=col[now];
            if (cc==z)
            {
                puts("Success.");
                continue;
            }
            if (du[z][x]==2||du[z][y]==2)
            {
                puts("Error 1.");
                continue;
            }
            if (find(z,x)==find(z,y))
            {
                puts("Error 2.");
                continue;
            }
            cut(cc,x,y);
            link(z,x,y);
            --du[cc][x];--du[cc][y];
            ++du[z][x];++du[z][y];
            col[now]=z;ext[now]=1;
            puts("Success.");
        }
        else if (opt==2)
        {
            int cc,x,y;scanf("%d%d%d",&cc,&x,&y);
            if (x>y) swap(x,y);
            if (find(cc,x)!=find(cc,y)) puts("-1");
            else
            {
                reverse(cc,x);
                access(cc,y);
                splay(cc,y);
                int ans=maxn[cc][y];
                printf("%d\n",ans);
            }
        }
    }
}