#include <bits/stdc++.h>
using namespace std;
#define LCBorz ios_base::sync_with_stdio(false); cin.tie(0);
#define int long long
#define all(x) x.begin(), x.end()
#define endl '\n'
const int N=200005;
const int INF=1e9;
vector<int> adj[N];
vector dp(4,vector<int>(N,INF));
vector v(N,0);
void dfs(int k,int pa){
vector<int> dp0,dp1;
int t0=0,t1=0;
for(int j:adj[k]){
if(j==pa)continue;
dfs(j,k);
t0+=dp[0][j];
dp0.push_back({dp[1][j]-dp[0][j]});
t1+=dp[2][j];
dp1.push_back({dp[3][j]-dp[2][j]});
}
sort(all(dp0));
sort(all(dp1));
int now=v[k]*2;
dp[now][k]=min(dp[now][k],t0);
for(int j:dp0){
now^=2;
t0+=j;
dp[now][k]=min(dp[now][k],t0);
}
now=(v[k]^1)*2+1;
dp[now][k]=min(dp[now][k],t1+1);
for(int j:dp1){
now^=2;
t1+=j;
dp[now][k]=min(dp[now][k],t1+1);
}
}
// change to 0
int32_t main() {
LCBorz;
int n;cin>>n;
for(int i=1;i<n;i++){
int a,b;cin>>a>>b;
adj[a].push_back(b);
adj[b].push_back(a);
}
for(int i=1;i<=n;i++){
cin>>v[i];
}
dfs(1,1);
int ans=min(dp[0][1],dp[1][1]);
if(ans==INF){
cout<<"impossible"<<endl;
}
else{
cout<<ans<<endl;
}
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
12888 KB |
Output is correct |
2 |
Correct |
5 ms |
12860 KB |
Output is correct |
3 |
Correct |
6 ms |
12728 KB |
Output is correct |
4 |
Correct |
6 ms |
12892 KB |
Output is correct |
5 |
Correct |
7 ms |
12888 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
12888 KB |
Output is correct |
2 |
Correct |
5 ms |
12860 KB |
Output is correct |
3 |
Correct |
6 ms |
12728 KB |
Output is correct |
4 |
Correct |
6 ms |
12892 KB |
Output is correct |
5 |
Correct |
7 ms |
12888 KB |
Output is correct |
6 |
Correct |
5 ms |
12892 KB |
Output is correct |
7 |
Correct |
5 ms |
12892 KB |
Output is correct |
8 |
Correct |
5 ms |
12892 KB |
Output is correct |
9 |
Correct |
5 ms |
12892 KB |
Output is correct |
10 |
Correct |
5 ms |
12768 KB |
Output is correct |
11 |
Correct |
5 ms |
12892 KB |
Output is correct |
12 |
Correct |
5 ms |
12892 KB |
Output is correct |
13 |
Correct |
5 ms |
12964 KB |
Output is correct |
14 |
Correct |
5 ms |
12892 KB |
Output is correct |
15 |
Correct |
6 ms |
12892 KB |
Output is correct |
16 |
Correct |
5 ms |
12892 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
48 ms |
33776 KB |
Output is correct |
2 |
Correct |
49 ms |
33332 KB |
Output is correct |
3 |
Correct |
50 ms |
33696 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
66 ms |
33780 KB |
Output is correct |
2 |
Correct |
47 ms |
33340 KB |
Output is correct |
3 |
Correct |
52 ms |
33844 KB |
Output is correct |
4 |
Correct |
42 ms |
16980 KB |
Output is correct |
5 |
Correct |
47 ms |
17276 KB |
Output is correct |
6 |
Correct |
52 ms |
17428 KB |
Output is correct |
7 |
Correct |
6 ms |
12888 KB |
Output is correct |
8 |
Correct |
18 ms |
14020 KB |
Output is correct |
9 |
Correct |
45 ms |
17204 KB |
Output is correct |
10 |
Correct |
47 ms |
17204 KB |
Output is correct |
11 |
Correct |
47 ms |
18560 KB |
Output is correct |
12 |
Correct |
60 ms |
18832 KB |
Output is correct |
13 |
Correct |
63 ms |
17428 KB |
Output is correct |
14 |
Correct |
55 ms |
17724 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
12888 KB |
Output is correct |
2 |
Correct |
5 ms |
12860 KB |
Output is correct |
3 |
Correct |
6 ms |
12728 KB |
Output is correct |
4 |
Correct |
6 ms |
12892 KB |
Output is correct |
5 |
Correct |
7 ms |
12888 KB |
Output is correct |
6 |
Correct |
5 ms |
12892 KB |
Output is correct |
7 |
Correct |
5 ms |
12892 KB |
Output is correct |
8 |
Correct |
5 ms |
12892 KB |
Output is correct |
9 |
Correct |
5 ms |
12892 KB |
Output is correct |
10 |
Correct |
5 ms |
12768 KB |
Output is correct |
11 |
Correct |
5 ms |
12892 KB |
Output is correct |
12 |
Correct |
5 ms |
12892 KB |
Output is correct |
13 |
Correct |
5 ms |
12964 KB |
Output is correct |
14 |
Correct |
5 ms |
12892 KB |
Output is correct |
15 |
Correct |
6 ms |
12892 KB |
Output is correct |
16 |
Correct |
5 ms |
12892 KB |
Output is correct |
17 |
Correct |
48 ms |
33776 KB |
Output is correct |
18 |
Correct |
49 ms |
33332 KB |
Output is correct |
19 |
Correct |
50 ms |
33696 KB |
Output is correct |
20 |
Correct |
66 ms |
33780 KB |
Output is correct |
21 |
Correct |
47 ms |
33340 KB |
Output is correct |
22 |
Correct |
52 ms |
33844 KB |
Output is correct |
23 |
Correct |
42 ms |
16980 KB |
Output is correct |
24 |
Correct |
47 ms |
17276 KB |
Output is correct |
25 |
Correct |
52 ms |
17428 KB |
Output is correct |
26 |
Correct |
6 ms |
12888 KB |
Output is correct |
27 |
Correct |
18 ms |
14020 KB |
Output is correct |
28 |
Correct |
45 ms |
17204 KB |
Output is correct |
29 |
Correct |
47 ms |
17204 KB |
Output is correct |
30 |
Correct |
47 ms |
18560 KB |
Output is correct |
31 |
Correct |
60 ms |
18832 KB |
Output is correct |
32 |
Correct |
63 ms |
17428 KB |
Output is correct |
33 |
Correct |
55 ms |
17724 KB |
Output is correct |
34 |
Correct |
5 ms |
12892 KB |
Output is correct |
35 |
Correct |
5 ms |
12892 KB |
Output is correct |
36 |
Correct |
5 ms |
12892 KB |
Output is correct |
37 |
Correct |
5 ms |
12892 KB |
Output is correct |
38 |
Correct |
5 ms |
12892 KB |
Output is correct |
39 |
Correct |
5 ms |
12892 KB |
Output is correct |
40 |
Correct |
5 ms |
12892 KB |
Output is correct |
41 |
Correct |
5 ms |
12892 KB |
Output is correct |
42 |
Correct |
5 ms |
12936 KB |
Output is correct |
43 |
Correct |
5 ms |
12892 KB |
Output is correct |
44 |
Correct |
5 ms |
12924 KB |
Output is correct |
45 |
Correct |
65 ms |
33584 KB |
Output is correct |
46 |
Correct |
54 ms |
33596 KB |
Output is correct |
47 |
Correct |
53 ms |
33896 KB |
Output is correct |
48 |
Correct |
47 ms |
16944 KB |
Output is correct |
49 |
Correct |
48 ms |
17456 KB |
Output is correct |
50 |
Correct |
47 ms |
17472 KB |
Output is correct |
51 |
Correct |
5 ms |
12888 KB |
Output is correct |
52 |
Correct |
18 ms |
13884 KB |
Output is correct |
53 |
Correct |
50 ms |
17296 KB |
Output is correct |
54 |
Correct |
62 ms |
17372 KB |
Output is correct |
55 |
Correct |
46 ms |
18484 KB |
Output is correct |
56 |
Correct |
47 ms |
18620 KB |
Output is correct |
57 |
Correct |
44 ms |
17472 KB |
Output is correct |
58 |
Correct |
68 ms |
17868 KB |
Output is correct |
59 |
Correct |
18 ms |
13884 KB |
Output is correct |
60 |
Correct |
43 ms |
16692 KB |
Output is correct |
61 |
Correct |
46 ms |
17200 KB |
Output is correct |
62 |
Correct |
47 ms |
17208 KB |
Output is correct |
63 |
Correct |
48 ms |
17344 KB |
Output is correct |
64 |
Correct |
47 ms |
17216 KB |
Output is correct |
65 |
Correct |
41 ms |
17732 KB |
Output is correct |
66 |
Correct |
43 ms |
17720 KB |
Output is correct |
67 |
Correct |
38 ms |
20360 KB |
Output is correct |
68 |
Correct |
38 ms |
20332 KB |
Output is correct |
69 |
Correct |
37 ms |
20284 KB |
Output is correct |
70 |
Correct |
37 ms |
20292 KB |
Output is correct |