#include<bits/stdc++.h>
using namespace std;
typedef long long llint;
typedef pair <int, int> pi;
const int MAXN = 100005;
int n;
llint dp[MAXN][2][2], par[MAXN], col[MAXN];
vector <int> v[MAXN];
void dfs (int x, int rod) {
par[x] = rod;
for (auto sus : v[x]) {
if (sus != rod) dfs(sus, x);
}
}
llint calc (int x, int f, int p) {
if (dp[x][f][p] != -1) return dp[x][f][p];
llint res = 1e9;
llint sum = 0, br = 0, mn = 1e9;
for (auto sus : v[x]) {
if (sus == par[x]) continue;
llint a = calc(sus, 0, f);
llint b = calc(sus, 1, f);
sum += min(a, b);
if (b <= a) br ^= 1;
mn = min(mn, abs(a - b));
}
if (br != f ^ p ^ col[x]) res = sum + mn; else res = sum;
res += f;
return dp[x][f][p] = res;
}
int main () {
ios_base::sync_with_stdio(false);
cin.tie(0);
memset(dp, -1, sizeof dp);
cin >> n;
for (int i = 0; i < n-1; i++) {
int a, b;
cin >> a >> b;
v[a].push_back(b);
v[b].push_back(a);
}
for (int i = 1; i <= n; i++) {
cin >> col[i];
}
dfs(1, 0);
llint sol = min(calc(1, 0, 0), calc(1, 1, 0));
if (sol >= 1e9) cout << "impossible"; else cout << sol;
return 0;
}
Compilation message
xanadu.cpp: In function 'llint calc(int, int, int)':
xanadu.cpp:33:12: warning: suggest parentheses around comparison in operand of '^' [-Wparentheses]
33 | if (br != f ^ p ^ col[x]) res = sum + mn; else res = sum;
| ~~~^~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
5708 KB |
Output is correct |
2 |
Correct |
3 ms |
5708 KB |
Output is correct |
3 |
Correct |
4 ms |
5800 KB |
Output is correct |
4 |
Correct |
4 ms |
5800 KB |
Output is correct |
5 |
Correct |
3 ms |
5708 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
5708 KB |
Output is correct |
2 |
Correct |
3 ms |
5708 KB |
Output is correct |
3 |
Correct |
4 ms |
5800 KB |
Output is correct |
4 |
Correct |
4 ms |
5800 KB |
Output is correct |
5 |
Correct |
3 ms |
5708 KB |
Output is correct |
6 |
Correct |
4 ms |
5708 KB |
Output is correct |
7 |
Correct |
4 ms |
5708 KB |
Output is correct |
8 |
Correct |
3 ms |
5708 KB |
Output is correct |
9 |
Correct |
3 ms |
5708 KB |
Output is correct |
10 |
Correct |
5 ms |
5704 KB |
Output is correct |
11 |
Correct |
4 ms |
5832 KB |
Output is correct |
12 |
Correct |
4 ms |
5804 KB |
Output is correct |
13 |
Correct |
3 ms |
5708 KB |
Output is correct |
14 |
Correct |
4 ms |
5708 KB |
Output is correct |
15 |
Correct |
5 ms |
5708 KB |
Output is correct |
16 |
Correct |
3 ms |
5708 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
103 ms |
24216 KB |
Output is correct |
2 |
Correct |
97 ms |
23944 KB |
Output is correct |
3 |
Correct |
139 ms |
24340 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
98 ms |
24132 KB |
Output is correct |
2 |
Correct |
95 ms |
23940 KB |
Output is correct |
3 |
Correct |
107 ms |
24328 KB |
Output is correct |
4 |
Correct |
89 ms |
11180 KB |
Output is correct |
5 |
Correct |
113 ms |
11588 KB |
Output is correct |
6 |
Correct |
125 ms |
11716 KB |
Output is correct |
7 |
Correct |
5 ms |
5804 KB |
Output is correct |
8 |
Correct |
41 ms |
7812 KB |
Output is correct |
9 |
Correct |
89 ms |
11804 KB |
Output is correct |
10 |
Correct |
119 ms |
11904 KB |
Output is correct |
11 |
Correct |
149 ms |
12716 KB |
Output is correct |
12 |
Correct |
122 ms |
12912 KB |
Output is correct |
13 |
Correct |
93 ms |
11452 KB |
Output is correct |
14 |
Correct |
107 ms |
11820 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
5708 KB |
Output is correct |
2 |
Correct |
3 ms |
5708 KB |
Output is correct |
3 |
Correct |
4 ms |
5800 KB |
Output is correct |
4 |
Correct |
4 ms |
5800 KB |
Output is correct |
5 |
Correct |
3 ms |
5708 KB |
Output is correct |
6 |
Correct |
4 ms |
5708 KB |
Output is correct |
7 |
Correct |
4 ms |
5708 KB |
Output is correct |
8 |
Correct |
3 ms |
5708 KB |
Output is correct |
9 |
Correct |
3 ms |
5708 KB |
Output is correct |
10 |
Correct |
5 ms |
5704 KB |
Output is correct |
11 |
Correct |
4 ms |
5832 KB |
Output is correct |
12 |
Correct |
4 ms |
5804 KB |
Output is correct |
13 |
Correct |
3 ms |
5708 KB |
Output is correct |
14 |
Correct |
4 ms |
5708 KB |
Output is correct |
15 |
Correct |
5 ms |
5708 KB |
Output is correct |
16 |
Correct |
3 ms |
5708 KB |
Output is correct |
17 |
Correct |
103 ms |
24216 KB |
Output is correct |
18 |
Correct |
97 ms |
23944 KB |
Output is correct |
19 |
Correct |
139 ms |
24340 KB |
Output is correct |
20 |
Correct |
98 ms |
24132 KB |
Output is correct |
21 |
Correct |
95 ms |
23940 KB |
Output is correct |
22 |
Correct |
107 ms |
24328 KB |
Output is correct |
23 |
Correct |
89 ms |
11180 KB |
Output is correct |
24 |
Correct |
113 ms |
11588 KB |
Output is correct |
25 |
Correct |
125 ms |
11716 KB |
Output is correct |
26 |
Correct |
5 ms |
5804 KB |
Output is correct |
27 |
Correct |
41 ms |
7812 KB |
Output is correct |
28 |
Correct |
89 ms |
11804 KB |
Output is correct |
29 |
Correct |
119 ms |
11904 KB |
Output is correct |
30 |
Correct |
149 ms |
12716 KB |
Output is correct |
31 |
Correct |
122 ms |
12912 KB |
Output is correct |
32 |
Correct |
93 ms |
11452 KB |
Output is correct |
33 |
Correct |
107 ms |
11820 KB |
Output is correct |
34 |
Correct |
5 ms |
5724 KB |
Output is correct |
35 |
Correct |
4 ms |
5800 KB |
Output is correct |
36 |
Correct |
5 ms |
5800 KB |
Output is correct |
37 |
Correct |
4 ms |
5708 KB |
Output is correct |
38 |
Correct |
4 ms |
5796 KB |
Output is correct |
39 |
Correct |
4 ms |
5760 KB |
Output is correct |
40 |
Correct |
4 ms |
5708 KB |
Output is correct |
41 |
Correct |
4 ms |
5708 KB |
Output is correct |
42 |
Correct |
5 ms |
5800 KB |
Output is correct |
43 |
Correct |
4 ms |
5700 KB |
Output is correct |
44 |
Correct |
4 ms |
5708 KB |
Output is correct |
45 |
Correct |
84 ms |
24260 KB |
Output is correct |
46 |
Correct |
89 ms |
23920 KB |
Output is correct |
47 |
Correct |
85 ms |
24340 KB |
Output is correct |
48 |
Correct |
95 ms |
11176 KB |
Output is correct |
49 |
Correct |
114 ms |
11820 KB |
Output is correct |
50 |
Correct |
103 ms |
11848 KB |
Output is correct |
51 |
Correct |
4 ms |
5932 KB |
Output is correct |
52 |
Correct |
36 ms |
7848 KB |
Output is correct |
53 |
Correct |
146 ms |
11784 KB |
Output is correct |
54 |
Correct |
90 ms |
11968 KB |
Output is correct |
55 |
Correct |
93 ms |
12708 KB |
Output is correct |
56 |
Correct |
114 ms |
13124 KB |
Output is correct |
57 |
Correct |
95 ms |
11440 KB |
Output is correct |
58 |
Correct |
112 ms |
11736 KB |
Output is correct |
59 |
Correct |
26 ms |
7792 KB |
Output is correct |
60 |
Correct |
94 ms |
11204 KB |
Output is correct |
61 |
Correct |
100 ms |
11796 KB |
Output is correct |
62 |
Correct |
129 ms |
11900 KB |
Output is correct |
63 |
Correct |
104 ms |
11844 KB |
Output is correct |
64 |
Correct |
88 ms |
11896 KB |
Output is correct |
65 |
Correct |
87 ms |
12452 KB |
Output is correct |
66 |
Correct |
86 ms |
12364 KB |
Output is correct |
67 |
Correct |
77 ms |
12200 KB |
Output is correct |
68 |
Correct |
92 ms |
12176 KB |
Output is correct |
69 |
Correct |
84 ms |
12124 KB |
Output is correct |
70 |
Correct |
89 ms |
12188 KB |
Output is correct |