[CF805E] Ice cream coloring（dfs，贪心）

题目链接：http://codeforces.com/contest/805/problem/E

题意：一棵树有n个点，每一个点相当于一个有si个点的完全图。现在希望给每一个点染色，并且整个图上相邻两个不同类型的冰淇淋颜色不一样，同时，假如确定了一个颜色，那么整棵树上的同类型冰淇淋都是一个颜色。问一个可行的染色方案。

思路：直接从一个点开始dfs，遇到一个节点遍历这个节点的子图，看看哪个颜色用过，哪个颜色没用。同时也看下哪个点之前有颜色了，哪个点之前没有颜色。然后就从左到右扫没用过的颜色，直接染色即可。

注意每次递归的时候，vis数组memset会TLE，所以事先存好出现的颜色，之后挨个点置零。

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 300300;
int n, m, rcnt;
int ret[maxn];
bool vis[maxn];
vector<int> ice[maxn];
vector<int> G[maxn];

void dfs(int u, int p) {
    vector<int> tmp;
    for(int i = 0; i < ice[u].size(); i++) {
        int x = ice[u][i];
        if(ret[x]) {
            tmp.push_back(x);
            vis[ret[x]] = 1;
        }
    }
    int cnt = 1;
    for(int i = 0; i < ice[u].size(); i++) {
        int x = ice[u][i];
        if(ret[x]) continue;
        while(vis[cnt]) cnt++;
        ret[x] = cnt++;
    }
    for(int i = 0; i < tmp.size(); i++) vis[ret[tmp[i]]] = 0;
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if(v == p) continue;
        dfs(v, u);
    }
}

int main() {
    // freopen("in", "r", stdin);
    int u, v;
    while(~scanf("%d%d",&n,&m)) {
        memset(ret, 0, sizeof(ret));
        memset(vis, 0, sizeof(vis));
        rcnt = 0;
        for(int i = 1; i <= n; i++) {
            int cnt, c;
            ice[i].clear(); G[i].clear();
            scanf("%d", &cnt);
            for(int j = 0; j < cnt; j++) {
                scanf("%d", &c);
                ice[i].push_back(c);
            }
        }
        for(int i = 0; i < n-1; i++) {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1, 0);
        int p = 0;
        for(int i = 1; i <= m; i++) {
            if(!ret[i]) ret[i] = 1;
            p = max(p, ret[i]);
        }
        printf("%d\n", p);
        for(int i = 1; i <= m; i++) {
            printf("%d%c", ret[i], i==m?'\n':' ');
        }
    }
    return 0;
}