1. 程式人生 > >CodeForces 804C Ice cream coloring

CodeForces 804C Ice cream coloring

problems pen dfs img define int main ems pri

Ice cream coloring

題解:

這個題目中最關鍵的一句話是, 把任意一種類型的冰激淩所在的所有節點拿下來之後,這些節點是一個連通圖(樹)。

所以就不會存在多個set+起來之後是一個新的完全圖。

所以只要直接去做就好了。

對於每個節點來說,染色。

代碼:

技術分享圖片
#include<bits/stdc++.h>
using namespace std;
#define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);
#define LL long long
#define
ULL unsigned LL #define fi first #define se second #define pb push_back #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 #define lch(x) tr[x].son[0] #define rch(x) tr[x].son[1] #define max3(a,b,c) max(a,max(b,c)) #define min3(a,b,c) min(a,min(b,c)) typedef pair<int,int> pll;
const int inf = 0x3f3f3f3f; const int _inf = 0xc0c0c0c0; const LL INF = 0x3f3f3f3f3f3f3f3f; const LL _INF = 0xc0c0c0c0c0c0c0c0; const LL mod = (int)1e9+7; const int N = 3e5 + 100; vector<int> vc[N], e[N]; int col[N]; int vis[N]; int mx = 0; void dfs(int o, int u){ for(int v : vc[u]){ vis[col[v]]
++; } int b = 1; for(int v : vc[u]){ if(col[v]) continue; while(vis[b]) ++b; vis[b] = 1; col[v] = b; } mx = max(mx, b); for(int v : vc[u]){ vis[col[v]] = 0; } for(int v : e[u]){ if(v == o) continue; dfs(u, v); } } int main(){ int n, m; scanf("%d%d", &n, &m); for(int i = 1; i <= n; ++i){ int si, tv; scanf("%d", &si); for(int j = 1; j <= si; ++j){ scanf("%d", &tv); vc[i].pb(tv); } } int u, v; for(int i = 1; i < n; ++i){ scanf("%d%d", &u, &v); e[u].pb(v); e[v].pb(u); } dfs(0, 1); printf("%d\n", mx); for(int i = 1; i <= m; ++i){ if(!col[i]) col[i] = 1; printf("%d%c", col[i], " \n"[i==m]); } return 0; }
View Code

CodeForces 804C Ice cream coloring