Rainbow Roads

Rainbow Roads

题意：给一棵n个结点的树，每条边都有一种颜色，要求找出所有的good点，good点定义：从这个点出发，到其他任意结点的简单路径（最短路径），相邻的两条边颜色都不同。

 

 题解看这里：here

不过我想了一下\(vis\)设成2就可以了

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define endl '\n'
const int maxn = 5e5+10;

struct node{
    int to,nxt,col;
}e[maxn<<1];

int n,head[maxn],tot,c[maxn],vis[maxn];
int ans[maxn],cnt;

void init(){
    memset(head,-1,sizeof(head));
    tot=0;
}

void addedge(int u,int v,int w){
    e[tot].to=v; e[tot].nxt=head[u]; e[tot].col=w; head[u]=tot++;
    e[tot].to=u; e[tot].nxt=head[v]; e[tot].col=w; head[v]=tot++;
}

void dfs(int fa,int now){
    if( vis[now]>2 ) return ;
    vis[now]++;
    for(int i=head[now];~i;i=e[i].nxt){
        if( e[i].to!=fa )
            dfs(now,e[i].to);
    }
}

int main()
{
    init();
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        addedge(a,b,c);
    }

    for(int i=1;i<=n;i++){
        for(int j=head[i];~j;j=e[j].nxt){
            c[e[j].col]++;
        }
        for(int j=head[i];~j;j=e[j].nxt){
            if( c[e[j].col]>1 ){
                dfs(i,e[j].to);
            }
        }
        for(int j=head[i];~j;j=e[j].nxt){
            c[e[j].col]--;
        }
    }

    for(int i=1;i<=n;i++){
        if( !vis[i] ) ans[++cnt]=i;
    }
    printf("%d\n",cnt);
    for(int i=1;i<=cnt;i++){
        printf("%d\n",ans[i]);
    }

    return 0;
}