#include #include #include using namespace std; template inline bool chmin(T &a, const S &b) { return (a > b ? a = b, true : false); } constexpr int INF = 1e9; int main() { int n; cin >> n; vector> g(n); for (int i = 0; i < n - 1; i++) { int u, v; cin >> u >> v; u--; v--; g[u].emplace_back(v); g[v].emplace_back(u); } vector a(n); for (int i = 0; i < n; i++) cin >> a[i]; vector> dp(n); // 0: contract to A_v (0 or 2) // 1: contract to 1, root is uncovered / root can be used in matching // 2: contract to 1, root is covered / root can be used in matching // 3: contract to 1, root is covered / root cannot be used in matching auto rec = [&] (this auto self, int u, int p) -> void { array cur; cur.fill(INF); if (a[u] == 1) cur[3] = 0; else cur[0] = cur[1] = 0; for (int v : g[u]) if (v != p) { self(v, u); array nxt; nxt.fill(INF); // cut u-v int mn = INF; for (int i : {0, 2, 3}) chmin(mn, dp[v][i] + 1); for (int i = 0; i < 4; i++) chmin(nxt[i], cur[i] + mn); auto calc = [&](int s_u, int s_v, int s_nxt) { chmin(nxt[s_nxt], cur[s_u] + dp[v][s_v]); }; // preserve u-v if (a[u] == 1 and a[v] == 1) { // 1-1 calc(3, 3, 3); } else if (a[u] == 1) { // 1-(0,2) calc(3, 3, 3); calc(3, 2, 3); } else if (a[v] == 1) { // (0,2)-1 calc(0, 3, 1); calc(1, 3, 1); calc(2, 3, 2); calc(3, 3, 3); } else if (a[u] == a[v]) { // 0-0 or 2-2 calc(0, 0, 0); for (int i : {0, 1}) for (int j : {0, 1}) calc(i, j, 1); for (int i : {0, 1}) for (int j : {2, 3}) calc(i, j, 2); for (int i : {2, 3}) for (int j : {0, 1, 2, 3}) calc(i, j, i); } else { // 0-2 or 2-0 // no match for (int i : {0, 1, 2, 3}) for (int j : {2,3}) { int k = (i == 0 ? 1 : i); calc(i, j, k); } // match for (int i : {0, 1, 2}) for (int j : {0, 1, 2}) { calc(i, j, 3); } } swap(cur, nxt); } dp[u] = cur; }; rec(0, -1); int cuts = n; for (int i : {0, 2, 3}) chmin(cuts, dp[0][i]); int ans = n - 1 - cuts; cout << ans << "\n"; }