E - Rainbow Roads （ dfs+思维 ）

E - Rainbow Roads （ dfs+思维 ）

 题目链接 Kattis - rainbowroads

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

Sample Input 1
8
1 3 1
2 3 1
3 4 3
4 5 4
5 6 3
6 7 2
6 8 2

Sample Output 1
4
3
4
5
6

思路：考虑不满足条件的情况，简化为三个点u，v，w， <u,v>,  <u,w>两条件为同一个颜色。那么从u出发到v，以及所有能到达的点都不满足good点的定义，从u出发到w同理，注意每个点可能会走过好多遍，直接设置大一点结束就OK了。

代码：

#include <bits/stdc++.h>

using namespace std;

const int maxn = 2e5+10;
struct node {
    int to,nxt,w;
}e[maxn*2];
int head[maxn],cnt;
int via[maxn],n;
int color[maxn];

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

void dfs( int x, int pre )
{
    if ( via[x]>10 ) return ;
    via[x]++;
    for ( int i=head[x]; i!=-1; i=e[i].nxt ) {
        int to = e[i].to;
        if ( to==pre ) continue;
        dfs(to,x);
    }
}

int main()
{
    memset(head,-1,sizeof(head));cnt=0;
    cin >> n;
    for ( int i=1; i<n; i++ ) {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        addage(u,v,w);
        addage(v,u,w);
    }
    for ( int i=1; i<=n; i++ ) {
        //memset(color,0,sizeof(color)); // 太浪费时间了，下面其实只改了三两个点，但整个数组有2e5个点
        for ( int j=head[i]; j!=-1; j=e[j].nxt ) {
            color[e[j].w]++;
        }
        for ( int j=head[i]; j!=-1; j=e[j].nxt ) {
            if ( color[e[j].w]>1 ) {
                int to = e[j].to;
                dfs(to,i);
            }
        }
        for ( int j=head[i]; j!=-1; j=e[j].nxt ) {
            color[e[j].w]--;
        }
    }
    vector<int> v;
    for ( int i=1; i<=n; i++ ) if ( via[i]==0 ) v.push_back(i);
    cout << v.size() << endl;
    for ( int i=0; i<v.size(); i++ ) {
        printf("%d\n",v[i]);
    }

    return 0;
}