// #pragma GCC target ("avx") // #pragma GCC optimize("Ofast") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #ifdef MARC_LOCAL #define _GLIBCXX_DEBUG #endif #ifdef ATCODER #include using namespace atcoder; #endif #include using namespace std; //#define int long long using ll = long long; using vll = vector; using vvll = vector; using vvvll = vector; using vvvvll = vector; using vi = vector; using vvi = vector; using vvvi = vector; using vvvvi = vector; using ld = long double; using vld = vector; using vvld = vector; using vd = vector; using vc = vector; using vvc = vector; using vs = vector; using vb = vector; using vvb = vector; using vvvb = vector; using pii = pair; using pcc = pair; using pll = pair; using pli = pair; using pdd = pair; using pldld = pair; using vpii = vector; using vvpii = vector; using vpll = vector; using vvpll = vector; using vpldld = vector; using ui = unsigned int; using ull = unsigned long long; using i128 = __int128; using f128 = __float128; template using pqg = priority_queue, greater>; #define vec(type, name, ...) vector name(__VA_ARGS__) #define vv(type, name, h, ...) vector > name(h, vector(__VA_ARGS__)) #define vvv(type, name, h, w, ...) vector > > name(h, vector >(w, vector(__VA_ARGS__))) #define vvvv(type, name, a, b, c, ...) vector > > > name(a, vector > >(b, vector >(c, vector(__VA_ARGS__)))) #define overload4(a,b,c,d,name,...) name #define rep1(n) for(ll i = 0; i < (ll)n; i++) #define rep2(i,n) for(ll i = 0; i < (ll)n; i++) #define rep3(i,a,b) for (ll i = a; i < (ll)b; i++) #define rep4(i,a,b,c) for (ll i = a; i < (ll)b; i += (ll)c) #define rep(...) overload4(__VA_ARGS__,rep4,rep3,rep2,rep1)(__VA_ARGS__) #define repback1(n) for (ll i = (ll)n-1; i >= 0; i--) #define repback2(i,n) for (ll i = (ll)n-1; i >= 0; i--) #define repback3(i,a,b) for (ll i = (ll)b-1; i >= (ll)a; i--) #define repback4(i,a,b,c) for (ll i = (ll)b-1; i >= (ll)a; i -= (ll)c) #define repback(...) overload4(__VA_ARGS__,repback4,repback3,repback2,repback1)(__VA_ARGS__) #define all(x) (x).begin(), (x).end() #define include(y, x, H, W) (0 <= (y) && (y) < (H) && 0 <= (x) && (x) < (W)) #define inrange(x, down, up) ((down) <= (x) && (x) <= (up)) #define pb push_back #define eb emplace_back #define fir first #define sec second #define TIMER_START TIME_START = clock() #ifdef MARC_LOCAL // https://trap.jp/post/1224/ #define debug1(x) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << endl #define debug2(x, y) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << ", " << (#y) << ": " << (y) << endl #define debug3(x, y, z) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << ", " << (#y) << ": " << (y) << ", " << (#z) << ": " << (z) << endl #define debug4(x, y, z, w) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << ", " << (#y) << ": " << (y) << ", " << (#z) << ": " << (z) << ", " << (#w) << ": " << (w) << endl #define debug5(x, y, z, w, v) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << ", " << (#y) << ": " << (y) << ", " << (#z) << ": " << (z) << ", " << (#w) << ": " << (w) << ", " << (#v) << ": " << (v) << endl #define debug6(x, y, z, w, v, u) cout << "debug L" << __LINE__ <<": " << (#x) << ": " << (x) << ", " << (#y) << ": " << (y) << ", " << (#z) << ": " << (z) << ", " << (#w) << ": " << (w) << ", " << (#v) << ": " << (v) << ", " << (#u) << ": " << (u) << endl #define overload6(a, b, c, d, e, f, g,...) g #define debug(...) overload6(__VA_ARGS__, debug6, debug5, debug4, debug3, debug2, debug1)(__VA_ARGS__) #define debuga cerr << __LINE__ << endl #define debugnl cout << endl #define Case(i) cout << "Case #" << (i) << ": " #define TIMECHECK cerr << 1000.0 * static_cast(clock() - TIME_START) / CLOCKS_PER_SEC << "ms" << endl #define LOCAL 1 #else #define debug1(x) void(0) #define debug2(x, y) void(0) #define debug3(x, y, z) void(0) #define debug4(x, y, z, w) void(0) #define debug5(x, y, z, w, v) void(0) #define debug6(x, y, z, w, v, u) void(0) #define debug(...) void(0) #define debuga void(0) #define debugnl void(0) #define Case(i) void(0) #define TIMECHECK void(0) #define LOCAL 0 #endif //mt19937_64 rng(0); mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()); clock_t TIME_START; const long double pi = 3.141592653589793238462643383279L; const long long INFL = 1ll<<61; const long long INFLMAX = numeric_limits< long long >::max(); // 9223372036854775807 const int INF = 1000000000; const int INFMAX = numeric_limits< int >::max(); // 2147483647 const long double INFD = numeric_limits::infinity(); const long double EPS = 1e-10; const int mod1 = 1000000007; const int mod2 = 998244353; const vi dx1 = {1,0,-1,0}; const vi dy1 = {0,1,0,-1}; const vi dx2 = {0, 1, 1, 1, 0, -1, -1, -1}; const vi dy2 = {1, 1, 0, -1, -1, -1, 0, 1}; constexpr char nl = '\n'; constexpr char bl = ' '; std::ostream &operator<<(std::ostream &dest, __int128_t value) { std::ostream::sentry s(dest); if (s) {__uint128_t tmp = value < 0 ? -value : value; char buffer[128]; char *d = std::end(buffer); do {--d;*d = "0123456789"[tmp % 10];tmp /= 10;} while (tmp != 0);if (value < 0) {--d;*d = '-';}int len = std::end(buffer) - d;if (dest.rdbuf()->sputn(d, len) != len) {dest.setstate(std::ios_base::badbit);}}return dest; } __int128 parse(string &s) { __int128 ret = 0; for (int i = 0; i < (int)s.length(); i++) if ('0' <= s[i] && s[i] <= '9') ret = 10 * ret + s[i] - '0'; return ret; } // for __int128 (https://kenkoooo.hatenablog.com/entry/2016/11/30/163533) template ostream& operator << (ostream& os, vector& vec) { os << "["; for (int i = 0; i<(int)vec.size(); i++) { os << vec[i] << (i + 1 == (int)vec.size() ? "" : ", "); } os << "]"; return os; } /// vector 出力 template ostream& operator << (ostream& os, pair& pair_var) { os << "(" << pair_var.first << ", " << pair_var.second << ")"; return os; } // pair 出力 template ostream& operator << (ostream& os, map& map_var) { os << "{"; for (auto itr = map_var.begin(); itr != map_var.end(); itr++) { os << "(" << itr->first << ", " << itr->second << ")"; itr++; if(itr != map_var.end()) os << ", "; itr--; } os << "}"; return os; } // map出力 template ostream& operator << (ostream& os, set& set_var) { os << "{"; for (auto itr = set_var.begin(); itr != set_var.end(); itr++) { os << *itr; ++itr; if(itr != set_var.end()) os << ", "; itr--; } os << "}"; return os; } /// set 出力 bool equals(long double a, long double b) { return fabsl(a - b) < EPS; } template int sign(T a) { return equals(a, 0) ? 0 : a > 0 ? 1 : -1; } long double radian_to_degree(long double r) { return (r * 180.0 / pi); } long double degree_to_radian(long double d) { return (d * pi / 180.0); } int popcnt(unsigned long long a){ return __builtin_popcountll(a); } // ll は 64bit対応! int MSB1(unsigned long long x) { return (x == 0 ? -1 : 63 - __builtin_clzll(x)); } // MSB(1), 0-indexed ( (0, 1, 2, 3, 4) -> (-1, 0, 1, 1, 2) ) int LSB1(unsigned long long x) { return (x == 0 ? -1 : __builtin_ctzll(x)); } // LSB(1), 0-indexed ( (0, 1, 2, 3, 4) -> (-1, 0, 1, 0, 2) ) long long maskbit(int n) { return (1LL << n) - 1; } bool bit_is1(long long x, int i) { return ((x>>i) & 1); } string charrep(int n, char c) { return std::string(n, c); } template T square(T x) { return (x) * (x); } templatevoid UNIQUE(T& A) {sort(all(A)); A.erase(unique(all(A)), A.end());} templatebool chmax(T& a, const T& b) { if (a < b) { a = b; return 1; } return 0; } templatebool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; } template bool chmin(T& a, const U& b){ return chmin(a, (T)b); } template bool chmax(T& a, const U& b){ return chmax(a, (T)b); } void YESNO(bool b) {if(b){cout<<"YES"<<'\n';} else{cout<<"NO"<<'\n';}} void YES() {YESNO(true);} void NO() {YESNO(false);} void yesno(bool b) {if(b){cout<<"yes"<<'\n';} else{cout<<"no"<<'\n';}} void yes() {yesno(true);} void no() {yesno(false);} void YesNo(bool b) {if(b){cout<<"Yes"<<'\n';} else{cout<<"No"<<'\n';}} void Yes() {YesNo(true);} void No() {YesNo(false);} void POSIMPOS(bool b) {if(b){cout<<"POSSIBLE"<<'\n';} else{cout<<"IMPOSSIBLE"<<'\n';}} void PosImpos(bool b) {if(b){cout<<"Possible"<<'\n';} else{cout<<"Impossible"<<'\n';}} void posimpos(bool b) {if(b){cout<<"possible"<<'\n';} else{cout<<"impossible"<<'\n';}} void FIRSEC(bool b) {if(b){cout<<"FIRST"<<'\n';} else{cout<<"SECOND"<<'\n';}} void firsec(bool b) {if(b){cout<<"first"<<'\n';} else{cout<<"second"<<'\n';}} void FirSec(bool b) {if(b){cout<<"First"<<'\n';} else{cout<<"Second"<<'\n';}} void AliBob(bool b) {if(b){cout<<"Alice"<<'\n';} else{cout<<"Bob"<<'\n';}} void TakAok(bool b) {if(b){cout<<"Takahashi"<<'\n';} else{cout<<"Aoki"<<'\n';}} int GetTime() {return 1000.0*static_cast(clock() - TIME_START) / CLOCKS_PER_SEC;} ll myRand(ll B) {return (unsigned long long)rng() % B;} template void print(const T& x, const char endch = '\n') { cout << x << endch; } template T ceil_div(T x, T y) { assert(y); return (x > 0 ? (x + y - 1) / y : x / y); } template T floor_div(T x, T y) { assert(y); return (x > 0 ? x / y : (x - y + 1) / y); } template pair divmod(T x, T y) { T q = floor_div(x, y); return {q, x - q * y}; } /// (q, r) s.t. x = q*y + r ll GCD(ll a, ll b) { if(a < b) swap(a, b); if(b == 0) return a; if(a%b == 0) return b; else return GCD(b, a%b); } ll LCM(ll a, ll b) { assert(GCD(a,b) != 0); return a / GCD(a, b) * b; } ll MOD(ll &x, const ll P) { ll ret = x%P; if(ret < 0) ret += P; return x = ret; } /// x % P を非負整数に直す ll mpow(ll x, ll n, const ll mod) { x %= mod; ll ret = 1; while(n > 0) { if(n & 1) ret = ret * x % mod; x = x * x % mod; n >>= 1; } return ret; } /// x^n % mod を計算 ll lpow(ll x, ll n) { ll ret = 1; while(n > 0){ if(n & 1) ret = ret * x; x = x * x; n >>= 1; } return ret; } /// x^nを計算 string toBinary(ll n) { if(n == 0) return "0"; assert(n > 0); string ret; while (n != 0){ ret += ( (n & 1) == 1 ? '1' : '0' ); n >>= 1; } reverse(ret.begin(), ret.end()); return ret; } /// 10進数(long long) -> 2進数(string)への変換 ll toDecimal(string S) { ll ret = 0; for(int i = 0; i < (int)S.size(); i++){ ret *= 2LL; if(S[i] == '1') ret += 1; } return ret; } /// 2進数(string) → 10進数(long long)への変換 int ceil_pow2(ll n) { int x = 0; while ((1ll << x) < n) x++; return x;} /// return minimum non-negative `x` s.t. `n <= 2**x` int floor_pow2(ll n) { int x = 0; while ((1ll << (x+1)) <= n) x++; return x;} /// return maximum non-negative `x` s.t. `n >= 2**x` // ############################ // # # // # C O D E S T A R T # // # # // ############################ /// Union-Find struct unionfind{ vector par, siz; unionfind(int n) : par(n, -1), siz(n, 1) {} // 根を求める int root(int x) { if (par[x] == -1) return x; else return par[x] = root(par[x]); } // xとyの根(グループ)が一致するかどうか bool issame(int x, int y){ return root(x) == root(y); } // xとyのグループの併合 bool unite(int x, int y){ x = root(x); y = root(y); if (x == y) return false; if (siz[x] < siz[y]) swap(x,y); par[y] = x; siz[x] += siz[y]; return true; } // xを含むグループのサイズ int size(int x){ return siz[root(x)]; } }; void solve() { ll N; cin >> N; vvi G(N); rep(i,N-1) { int u,v; cin >> u >> v; u--; v--; G[u].pb(v); G[v].pb(u); } vi A(N); rep(i,N) cin >> A[i]; // vvi dp(N,vi(3,INF)); // auto dfs = [&](auto dfs, int v, int p)->void{ // for(auto x:G[v]) { // if(x==p) continue; // dfs(dfs,x,v); // } // int ret = 1; // for(auto x:G[v]) { // if(x==p) continue; // int mn = INF; // rep(j,3) { // if(dp[x][j]==INF) continue; // if(j==A[v]) chmin(mn,dp[x][j]-1); // else chmin(mn,dp[x][j]); // } // ret += mn; // } // dp[v][A[v]] = ret; // ret = 1; // int ng = 0; // if(A[v]==0) { // bool ok = false; // for(auto x:G[v]) { // if(x==p) continue; // if(dp[x][2] tmp; // rep(j,3) { // if(dp[x][j]second.size()==1) { // if(tmp.begin()->second[0]==2) { // if(used) ng++; // else used=true; // } // } // } // dp[v][1] = ret+ng; // } // } // else if(A[v]==2){ // bool ok = false; // for(auto x:G[v]) { // if(x==p) continue; // if(dp[x][0] tmp; // rep(j,3) { // if(dp[x][j]second.size()==1) { // if(tmp.begin()->second[0]==0) { // if(used) ng++; // else used=true; // } // } // } // dp[v][1] = ret+ng; // } // } // }; // dfs(dfs,0,-1); // debug(dp); // cout << N - *min_element(all(dp[0])) << nl; // vi ID(N); // rep(i,N) ID[i] = i; // unordered_set st; // vi vlst; // unionfind uf(N); // auto build = [&](auto build, int v, int p, int num)->void{ // if(A[v] == num) vlst.pb(v); // for(auto x:G[v]) { // if(x == p) continue; // dfs(dfs,) // } // }; // build(build,0,-1,A[0]); // unionfind uf(N); // rep(i,N) { // for(auto x:G[i]) { // if(A[x] == A[i]) uf.unite(x,i); // } // } // //int ans = 0; // vector > T(N); // vvi comp(N); // //vector > T(N); // //vi ones; // rep(i,N) { // comp[uf.root(i)].pb(i); // for(auto x:G[i]) { // if(uf.root(i)==uf.root(x)) continue; // T[uf.root(i)].insert(uf.root(x)); // T[uf.root(x)].insert(uf.root(i)); // } // //if(uf.root(i)!=i) ans++; // //if(uf.root(i)==i && A[i]==1) ones.pb(uf.root(i)); // } // rep(i,N) debug(T[i]); // debug(A); // debug(comp); //debug(ans); //debug(ones); // vvi dp(N,vi(3,INF)); // i以下の部分木、{1でない(自分と異なるも兼ねてる)、余裕ない1、余裕ある1}の最小頂点数 ->余裕ある=親-子の辺が使える // auto dfs = [&](auto dfs, int v, int p)->void{ // for(auto x:T[v]) { // if(x==p) continue; // dfs(dfs,x,v); // } // if(A[v]==0) { // } // else if(A[v]==1) { // int ret = 1; // for(auto x:T[v]) { // if(x==p) continue; // ret += min({dp[x][0],dp[x][1]-1,dp[x][2]-1}); // } // dp[v][2] = ret; // } // else { // } // }; vvi dp(N,vi(4,0)); // i以下の部分木、i含む同値成分が{最終的に1にならない、今後1になる、1に既に確定でできて辺を提供する余裕もある、1になってるが余裕がない} のときの縮約数 auto dfs = [&](auto dfs, int v, int p)->void{ vi cur(4,-INF); if(A[v]==1) { cur[2]=0; } else { cur[0]=0; cur[1]=0; } for(auto x:G[v]) { if(x==p) continue; dfs(dfs,x,v); vi nxt = cur; if(A[v]==A[x]) { rep(j,4) { rep(k,4) { //dp[x][j] と cur[k] -> nxt[?]に 辺単位でみる if(k==2||k==3) { if(j==0) continue; chmax(nxt[k],dp[x][j]+cur[k]+1); } if(k==1) { if(j==1) chmax(nxt[1],dp[x][j]+cur[k]+1); if(j==2) chmax(nxt[2],dp[x][j]+cur[k]+1); if(j==3) chmax(nxt[2],dp[x][j]+cur[k]+1); } if(k==0) { if(j==0) chmax(nxt[0],dp[x][j]+cur[k]+1); } } } } else if(A[v]!=A[x]){ if((A[v]==0&&A[x]==2)||(A[v]==2&&A[x]==0)) { rep(j,4) { rep(k,4) { if(j==0||k==0||j==3||k==3) continue; chmax(nxt[3],dp[x][j]+cur[k]+1); } } } //else { rep(j,4) { rep(k,4) { if(j==1) continue; if(j!=0&&k!=0) { chmax(nxt[k],dp[x][j]+cur[k]+1); } else { chmax(nxt[k],dp[x][j]+cur[k]); } } } //} } debug(v,x,nxt); swap(cur,nxt); } dp[v]=cur; // if(A[v]==0) { // vi ret(4,1); // for(auto x:T[v]) { // if(x==p) continue; // if(A[x]==0) { // rep(j,4) ret[j] += dp[x][j]-1; // } // else if(A[x]==1) { // dp[v][0] += dp[x][2]; // dp[v][1] += dp[x][2]; // dp[v][0] += dp[x][2]; // } // else { // } // } // } // else if(A[v]==1) { // int ret = 1; // for(auto x:T[v]) { // if(x==p) continue; // ret += min({dp[x][0],dp[x][1]-1,dp[x][2]-1,dp[x][3]-1}); // } // dp[v][2] = ret; // } // else { // } }; dfs(dfs,0,-1); debug(dp); int mx = 0; rep(j,4) { if(j==1) continue; chmax(mx,dp[0][j]); } cout << mx << nl; // rep(i,N) { // if(uf.root(i)==i) { // dfs(dfs,i,-1); // break; // } // } // int need = 0; // vi cnt(3,0); // auto dfs = [&](auto dfs, int v, int p)->int{ // if(v==-1) dfs(dfs,A[],v); // else { // for(auto x:T[i]) { // if(x == p) continue; // dfs(dfs,x,v); // } // if(v==-1 || A[v] == 1) { // cnt[0]=0; // cnt[1]=0; // cnt[2]=0; // } // } // auto dfs = [&](auto dfs, int v, int p)->int{ // vpii cs; // for(auto x:T[v]) { // if(x == p) continue; // cs.eb(dfs(dfs,x,v),x); // } // debug(v,A[v]); // for(auto [val,x]:cs) { // debug(val,x); // if((val==0&&A[v]==2)||(val==2&&A[v]==0)) { // //uf2.unite(x,v); // if(T[x].size()> tt; while(tt--){ solve(); } TIMECHECK; return 0; }