#include <bits/stdc++.h>
#define int ll
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define rng(i,c,n) for(int i=c;i<n;i++)
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define vec(...) vector<__VA_ARGS__>
#define _3D2ZDxo ios::sync_with_stdio(0),cin.tie(0)
typedef long long ll;
using pii=pair<int,int>;
using vi=vector<int>;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}
void chmin(int&a,int b){
a=min(a,b);
}
signed main(){
_3D2ZDxo;
int n;
cin>>n;
vec(vi) adj(n);
rep(i,n-1){
int u,v;
cin>>u>>v;
u-=1,v-=1;
adj[u].pb(v);
adj[v].pb(u);
}
vi a(n);
rep(i,n){
cin>>a[i];
}
rep(i,n){
a[i]^=1;
}
const int inf=1e9;
vec(vec(vi)) dp(n,vec(vi)(2,vi(2,inf)));
auto dfs=[&](auto self,int v,int par)->void{
for(auto u:adj[v]){
if(u==par) continue;
self(self,u,v);
}
vi fdp(2); // flip the root
fdp[a[v]^1]=1;
fdp[a[v]]=inf;
for(auto u:adj[v]){
if(u==par) continue;
vi nfdp(2,inf);
{
int t=0;
rep(t1,2){
rep(wit,2){
int nwit=wit^t1;
chmin(nfdp[nwit],fdp[wit]+dp[u][t][t1]);
}
}
}
fdp=nfdp;
}
vi gdp(2);
gdp[a[v]]=0;
gdp[a[v]^1]=inf;
for(auto u:adj[v]){
if(u==par) continue;
vi ngdp(2,inf);
{
int t=1;
rep(t1,2){
rep(wit,2){
int nwit=wit^t1;
chmin(ngdp[nwit],gdp[wit]+dp[u][t][t1]);
}
}
}
gdp=ngdp;
}
rep(t,2){
chmin(dp[v][t][1],fdp[t]);
chmin(dp[v][t][0],gdp[t]);
}
};
dfs(dfs,0,-1);
int ans=min(dp[0][1][0],dp[0][1][1]);
if(ans>=inf){
print("impossible");
}else{
print(ans);
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
324 KB |
Output is correct |
11 |
Correct |
1 ms |
316 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
212 KB |
Output is correct |
15 |
Correct |
1 ms |
320 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
101 ms |
43296 KB |
Output is correct |
2 |
Correct |
97 ms |
43924 KB |
Output is correct |
3 |
Correct |
86 ms |
44612 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
92 ms |
43144 KB |
Output is correct |
2 |
Correct |
98 ms |
43820 KB |
Output is correct |
3 |
Correct |
108 ms |
44708 KB |
Output is correct |
4 |
Correct |
99 ms |
21068 KB |
Output is correct |
5 |
Correct |
117 ms |
22948 KB |
Output is correct |
6 |
Correct |
127 ms |
23416 KB |
Output is correct |
7 |
Correct |
1 ms |
340 KB |
Output is correct |
8 |
Correct |
30 ms |
7916 KB |
Output is correct |
9 |
Correct |
116 ms |
22952 KB |
Output is correct |
10 |
Correct |
124 ms |
23452 KB |
Output is correct |
11 |
Correct |
104 ms |
24268 KB |
Output is correct |
12 |
Correct |
111 ms |
25160 KB |
Output is correct |
13 |
Correct |
115 ms |
22172 KB |
Output is correct |
14 |
Correct |
116 ms |
23736 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
324 KB |
Output is correct |
11 |
Correct |
1 ms |
316 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
212 KB |
Output is correct |
15 |
Correct |
1 ms |
320 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
101 ms |
43296 KB |
Output is correct |
18 |
Correct |
97 ms |
43924 KB |
Output is correct |
19 |
Correct |
86 ms |
44612 KB |
Output is correct |
20 |
Correct |
92 ms |
43144 KB |
Output is correct |
21 |
Correct |
98 ms |
43820 KB |
Output is correct |
22 |
Correct |
108 ms |
44708 KB |
Output is correct |
23 |
Correct |
99 ms |
21068 KB |
Output is correct |
24 |
Correct |
117 ms |
22948 KB |
Output is correct |
25 |
Correct |
127 ms |
23416 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
30 ms |
7916 KB |
Output is correct |
28 |
Correct |
116 ms |
22952 KB |
Output is correct |
29 |
Correct |
124 ms |
23452 KB |
Output is correct |
30 |
Correct |
104 ms |
24268 KB |
Output is correct |
31 |
Correct |
111 ms |
25160 KB |
Output is correct |
32 |
Correct |
115 ms |
22172 KB |
Output is correct |
33 |
Correct |
116 ms |
23736 KB |
Output is correct |
34 |
Correct |
1 ms |
320 KB |
Output is correct |
35 |
Correct |
1 ms |
212 KB |
Output is correct |
36 |
Correct |
1 ms |
212 KB |
Output is correct |
37 |
Correct |
1 ms |
212 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
1 ms |
212 KB |
Output is correct |
40 |
Correct |
1 ms |
212 KB |
Output is correct |
41 |
Correct |
1 ms |
212 KB |
Output is correct |
42 |
Correct |
1 ms |
212 KB |
Output is correct |
43 |
Correct |
1 ms |
320 KB |
Output is correct |
44 |
Correct |
0 ms |
212 KB |
Output is correct |
45 |
Correct |
93 ms |
44432 KB |
Output is correct |
46 |
Correct |
95 ms |
43852 KB |
Output is correct |
47 |
Correct |
105 ms |
44760 KB |
Output is correct |
48 |
Correct |
110 ms |
20996 KB |
Output is correct |
49 |
Correct |
118 ms |
22988 KB |
Output is correct |
50 |
Correct |
119 ms |
23364 KB |
Output is correct |
51 |
Correct |
1 ms |
340 KB |
Output is correct |
52 |
Correct |
30 ms |
8020 KB |
Output is correct |
53 |
Correct |
135 ms |
22972 KB |
Output is correct |
54 |
Correct |
112 ms |
23452 KB |
Output is correct |
55 |
Correct |
130 ms |
24248 KB |
Output is correct |
56 |
Correct |
136 ms |
25272 KB |
Output is correct |
57 |
Correct |
96 ms |
22160 KB |
Output is correct |
58 |
Correct |
115 ms |
23620 KB |
Output is correct |
59 |
Correct |
31 ms |
7904 KB |
Output is correct |
60 |
Correct |
106 ms |
20964 KB |
Output is correct |
61 |
Correct |
113 ms |
22944 KB |
Output is correct |
62 |
Correct |
115 ms |
23444 KB |
Output is correct |
63 |
Correct |
124 ms |
23448 KB |
Output is correct |
64 |
Correct |
119 ms |
23368 KB |
Output is correct |
65 |
Correct |
105 ms |
23912 KB |
Output is correct |
66 |
Correct |
102 ms |
23880 KB |
Output is correct |
67 |
Correct |
105 ms |
23488 KB |
Output is correct |
68 |
Correct |
109 ms |
23528 KB |
Output is correct |
69 |
Correct |
114 ms |
23496 KB |
Output is correct |
70 |
Correct |
141 ms |
23540 KB |
Output is correct |