#include <bits/stdc++.h>
using namespace std;
const long long MX=5e5+10;
const long long lg=30;
vector < long long > mas[MX];
long long prd[lg][MX];
long long tin[MX],tou[MX];
long long timer=0;
void DFSCNT(long long zar,long long mun)
{
timer++;
tin[zar]=timer;
prd[0][zar]=mun;
for(long long i=1;i<lg;i++)
{
prd[i][zar]=prd[i-1][prd[i-1][zar]];
}
for(auto u:mas[zar])
{
if(u!=mun)
{
DFSCNT(u,zar);
}
}
timer++;
tou[zar]=timer;
return;
}
long long ch_prd(long long pr,long long sn)
{
return (tin[pr]<=tin[sn] && tou[sn]<=tou[pr]);
}
long long lca(long long a,long long b)
{
if(ch_prd(a,b))
{
return a;
}
if(ch_prd(b,a))
{
return b;
}
for(long long i=lg-1;i>=0;i--)
{
if(!ch_prd(prd[i][a],b))
{
a=prd[i][a];
}
}
a=prd[0][a];
return a;
}
vector < long long > com[MX];
bool cmp(long long a,long long b)
{
return (tin[a]<tin[b]);
}
long long prfsum[MX],ans=0;
long long DFS2(long long zar,long long mun)
{
long long z=0;
for(auto u:mas[zar])
{
if(u!=mun)
{
z+=DFS2(u,zar);
prfsum[zar]+=prfsum[u];
}
}
if(zar==1)
{
ans+=(z==1);
}
else
{
if(prfsum[zar]==0)
{
ans+=(z==0);
z=1;
}
}
return z;
}
int main()
{
cin.tie(0);
ios_base::sync_with_stdio(0);
long long n,k;
cin>>n>>k;
long long a,b;
for(long long i=1;i<n;i++)
{
cin>>a>>b;
mas[a].push_back(b);
mas[b].push_back(a);
}
tin[0]=0;
DFSCNT(1,0);
timer++;
tou[0]=timer;
for(long long i=1;i<=n;i++)
{
cin>>a;
com[a].push_back(i);
}
for(long long i=1;i<=k;i++)
{
sort(com[i].begin(),com[i].end(),cmp);
long long sz=com[i].size();
for(long long j=0;j<sz;j++)
{
prfsum[com[i][j]]++;
prfsum[lca(com[i][j],com[i][(j+1)%sz])]--;
}
}
DFS2(1,0);
cout<<(ans+1)/2<<"\n";
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
91224 KB |
Output is correct |
2 |
Correct |
12 ms |
91228 KB |
Output is correct |
3 |
Correct |
12 ms |
91284 KB |
Output is correct |
4 |
Correct |
12 ms |
91324 KB |
Output is correct |
5 |
Correct |
13 ms |
91252 KB |
Output is correct |
6 |
Correct |
12 ms |
91332 KB |
Output is correct |
7 |
Correct |
12 ms |
91228 KB |
Output is correct |
8 |
Correct |
13 ms |
91228 KB |
Output is correct |
9 |
Correct |
12 ms |
91228 KB |
Output is correct |
10 |
Correct |
13 ms |
91224 KB |
Output is correct |
11 |
Correct |
12 ms |
91228 KB |
Output is correct |
12 |
Correct |
12 ms |
91228 KB |
Output is correct |
13 |
Correct |
12 ms |
91316 KB |
Output is correct |
14 |
Correct |
12 ms |
91228 KB |
Output is correct |
15 |
Correct |
12 ms |
91228 KB |
Output is correct |
16 |
Correct |
12 ms |
91228 KB |
Output is correct |
17 |
Correct |
12 ms |
91228 KB |
Output is correct |
18 |
Correct |
13 ms |
91228 KB |
Output is correct |
19 |
Correct |
12 ms |
91256 KB |
Output is correct |
20 |
Correct |
14 ms |
91204 KB |
Output is correct |
21 |
Correct |
12 ms |
91224 KB |
Output is correct |
22 |
Correct |
12 ms |
91228 KB |
Output is correct |
23 |
Correct |
12 ms |
91228 KB |
Output is correct |
24 |
Correct |
13 ms |
91228 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
91224 KB |
Output is correct |
2 |
Correct |
12 ms |
91228 KB |
Output is correct |
3 |
Correct |
12 ms |
91284 KB |
Output is correct |
4 |
Correct |
12 ms |
91324 KB |
Output is correct |
5 |
Correct |
13 ms |
91252 KB |
Output is correct |
6 |
Correct |
12 ms |
91332 KB |
Output is correct |
7 |
Correct |
12 ms |
91228 KB |
Output is correct |
8 |
Correct |
13 ms |
91228 KB |
Output is correct |
9 |
Correct |
12 ms |
91228 KB |
Output is correct |
10 |
Correct |
13 ms |
91224 KB |
Output is correct |
11 |
Correct |
12 ms |
91228 KB |
Output is correct |
12 |
Correct |
12 ms |
91228 KB |
Output is correct |
13 |
Correct |
12 ms |
91316 KB |
Output is correct |
14 |
Correct |
12 ms |
91228 KB |
Output is correct |
15 |
Correct |
12 ms |
91228 KB |
Output is correct |
16 |
Correct |
12 ms |
91228 KB |
Output is correct |
17 |
Correct |
12 ms |
91228 KB |
Output is correct |
18 |
Correct |
13 ms |
91228 KB |
Output is correct |
19 |
Correct |
12 ms |
91256 KB |
Output is correct |
20 |
Correct |
14 ms |
91204 KB |
Output is correct |
21 |
Correct |
12 ms |
91224 KB |
Output is correct |
22 |
Correct |
12 ms |
91228 KB |
Output is correct |
23 |
Correct |
12 ms |
91228 KB |
Output is correct |
24 |
Correct |
13 ms |
91228 KB |
Output is correct |
25 |
Correct |
12 ms |
91228 KB |
Output is correct |
26 |
Correct |
14 ms |
91364 KB |
Output is correct |
27 |
Correct |
14 ms |
91484 KB |
Output is correct |
28 |
Correct |
14 ms |
91740 KB |
Output is correct |
29 |
Correct |
14 ms |
91484 KB |
Output is correct |
30 |
Correct |
14 ms |
91572 KB |
Output is correct |
31 |
Correct |
12 ms |
91228 KB |
Output is correct |
32 |
Correct |
14 ms |
91740 KB |
Output is correct |
33 |
Correct |
12 ms |
91292 KB |
Output is correct |
34 |
Correct |
14 ms |
91436 KB |
Output is correct |
35 |
Correct |
14 ms |
91484 KB |
Output is correct |
36 |
Correct |
14 ms |
91484 KB |
Output is correct |
37 |
Correct |
14 ms |
91512 KB |
Output is correct |
38 |
Correct |
12 ms |
91228 KB |
Output is correct |
39 |
Correct |
15 ms |
91456 KB |
Output is correct |
40 |
Correct |
14 ms |
91480 KB |
Output is correct |
41 |
Correct |
14 ms |
91484 KB |
Output is correct |
42 |
Correct |
14 ms |
91516 KB |
Output is correct |
43 |
Correct |
14 ms |
91740 KB |
Output is correct |
44 |
Correct |
13 ms |
91224 KB |
Output is correct |
45 |
Correct |
14 ms |
91480 KB |
Output is correct |
46 |
Correct |
15 ms |
91480 KB |
Output is correct |
47 |
Correct |
13 ms |
91244 KB |
Output is correct |
48 |
Correct |
14 ms |
91484 KB |
Output is correct |
49 |
Correct |
14 ms |
91544 KB |
Output is correct |
50 |
Correct |
14 ms |
91484 KB |
Output is correct |
51 |
Correct |
14 ms |
91736 KB |
Output is correct |
52 |
Correct |
14 ms |
91484 KB |
Output is correct |
53 |
Correct |
15 ms |
91480 KB |
Output is correct |
54 |
Correct |
17 ms |
91480 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
91224 KB |
Output is correct |
2 |
Correct |
12 ms |
91228 KB |
Output is correct |
3 |
Correct |
12 ms |
91284 KB |
Output is correct |
4 |
Correct |
12 ms |
91324 KB |
Output is correct |
5 |
Correct |
13 ms |
91252 KB |
Output is correct |
6 |
Correct |
12 ms |
91332 KB |
Output is correct |
7 |
Correct |
12 ms |
91228 KB |
Output is correct |
8 |
Correct |
13 ms |
91228 KB |
Output is correct |
9 |
Correct |
12 ms |
91228 KB |
Output is correct |
10 |
Correct |
13 ms |
91224 KB |
Output is correct |
11 |
Correct |
12 ms |
91228 KB |
Output is correct |
12 |
Correct |
12 ms |
91228 KB |
Output is correct |
13 |
Correct |
12 ms |
91316 KB |
Output is correct |
14 |
Correct |
12 ms |
91228 KB |
Output is correct |
15 |
Correct |
12 ms |
91228 KB |
Output is correct |
16 |
Correct |
12 ms |
91228 KB |
Output is correct |
17 |
Correct |
12 ms |
91228 KB |
Output is correct |
18 |
Correct |
13 ms |
91228 KB |
Output is correct |
19 |
Correct |
12 ms |
91256 KB |
Output is correct |
20 |
Correct |
14 ms |
91204 KB |
Output is correct |
21 |
Correct |
12 ms |
91224 KB |
Output is correct |
22 |
Correct |
12 ms |
91228 KB |
Output is correct |
23 |
Correct |
12 ms |
91228 KB |
Output is correct |
24 |
Correct |
13 ms |
91228 KB |
Output is correct |
25 |
Correct |
12 ms |
91228 KB |
Output is correct |
26 |
Correct |
260 ms |
124668 KB |
Output is correct |
27 |
Correct |
265 ms |
124496 KB |
Output is correct |
28 |
Correct |
14 ms |
91568 KB |
Output is correct |
29 |
Correct |
12 ms |
91228 KB |
Output is correct |
30 |
Correct |
12 ms |
91332 KB |
Output is correct |
31 |
Correct |
189 ms |
124432 KB |
Output is correct |
32 |
Correct |
14 ms |
91484 KB |
Output is correct |
33 |
Correct |
93 ms |
130896 KB |
Output is correct |
34 |
Correct |
254 ms |
124336 KB |
Output is correct |
35 |
Correct |
14 ms |
91484 KB |
Output is correct |
36 |
Correct |
102 ms |
124464 KB |
Output is correct |
37 |
Correct |
14 ms |
91484 KB |
Output is correct |
38 |
Correct |
14 ms |
91520 KB |
Output is correct |
39 |
Correct |
263 ms |
124996 KB |
Output is correct |
40 |
Correct |
14 ms |
91740 KB |
Output is correct |
41 |
Correct |
197 ms |
124060 KB |
Output is correct |
42 |
Correct |
93 ms |
126036 KB |
Output is correct |
43 |
Correct |
12 ms |
91224 KB |
Output is correct |
44 |
Correct |
113 ms |
131052 KB |
Output is correct |
45 |
Correct |
235 ms |
128852 KB |
Output is correct |
46 |
Correct |
15 ms |
91484 KB |
Output is correct |
47 |
Correct |
14 ms |
91428 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
264 ms |
124872 KB |
Output is correct |
2 |
Correct |
76 ms |
126932 KB |
Output is correct |
3 |
Correct |
14 ms |
91484 KB |
Output is correct |
4 |
Correct |
13 ms |
91736 KB |
Output is correct |
5 |
Correct |
12 ms |
91228 KB |
Output is correct |
6 |
Correct |
12 ms |
91216 KB |
Output is correct |
7 |
Correct |
14 ms |
91484 KB |
Output is correct |
8 |
Correct |
237 ms |
125512 KB |
Output is correct |
9 |
Correct |
14 ms |
91484 KB |
Output is correct |
10 |
Correct |
180 ms |
124612 KB |
Output is correct |
11 |
Correct |
12 ms |
91228 KB |
Output is correct |
12 |
Correct |
92 ms |
124076 KB |
Output is correct |
13 |
Correct |
192 ms |
125396 KB |
Output is correct |
14 |
Correct |
91 ms |
126800 KB |
Output is correct |
15 |
Correct |
263 ms |
124580 KB |
Output is correct |
16 |
Correct |
14 ms |
91480 KB |
Output is correct |
17 |
Correct |
12 ms |
91228 KB |
Output is correct |
18 |
Correct |
128 ms |
126472 KB |
Output is correct |
19 |
Correct |
99 ms |
130212 KB |
Output is correct |
20 |
Correct |
14 ms |
91484 KB |
Output is correct |
21 |
Correct |
12 ms |
91228 KB |
Output is correct |
22 |
Correct |
232 ms |
125376 KB |
Output is correct |
23 |
Correct |
14 ms |
91480 KB |
Output is correct |
24 |
Correct |
112 ms |
124832 KB |
Output is correct |
25 |
Correct |
110 ms |
129876 KB |
Output is correct |
26 |
Correct |
14 ms |
91484 KB |
Output is correct |
27 |
Correct |
14 ms |
91484 KB |
Output is correct |
28 |
Correct |
14 ms |
91444 KB |
Output is correct |
29 |
Correct |
14 ms |
91484 KB |
Output is correct |
30 |
Correct |
15 ms |
91484 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
91224 KB |
Output is correct |
2 |
Correct |
12 ms |
91228 KB |
Output is correct |
3 |
Correct |
12 ms |
91284 KB |
Output is correct |
4 |
Correct |
12 ms |
91324 KB |
Output is correct |
5 |
Correct |
13 ms |
91252 KB |
Output is correct |
6 |
Correct |
12 ms |
91332 KB |
Output is correct |
7 |
Correct |
12 ms |
91228 KB |
Output is correct |
8 |
Correct |
13 ms |
91228 KB |
Output is correct |
9 |
Correct |
12 ms |
91228 KB |
Output is correct |
10 |
Correct |
13 ms |
91224 KB |
Output is correct |
11 |
Correct |
12 ms |
91228 KB |
Output is correct |
12 |
Correct |
12 ms |
91228 KB |
Output is correct |
13 |
Correct |
12 ms |
91316 KB |
Output is correct |
14 |
Correct |
12 ms |
91228 KB |
Output is correct |
15 |
Correct |
12 ms |
91228 KB |
Output is correct |
16 |
Correct |
12 ms |
91228 KB |
Output is correct |
17 |
Correct |
12 ms |
91228 KB |
Output is correct |
18 |
Correct |
13 ms |
91228 KB |
Output is correct |
19 |
Correct |
12 ms |
91256 KB |
Output is correct |
20 |
Correct |
14 ms |
91204 KB |
Output is correct |
21 |
Correct |
12 ms |
91224 KB |
Output is correct |
22 |
Correct |
12 ms |
91228 KB |
Output is correct |
23 |
Correct |
12 ms |
91228 KB |
Output is correct |
24 |
Correct |
13 ms |
91228 KB |
Output is correct |
25 |
Correct |
12 ms |
91228 KB |
Output is correct |
26 |
Correct |
14 ms |
91364 KB |
Output is correct |
27 |
Correct |
14 ms |
91484 KB |
Output is correct |
28 |
Correct |
14 ms |
91740 KB |
Output is correct |
29 |
Correct |
14 ms |
91484 KB |
Output is correct |
30 |
Correct |
14 ms |
91572 KB |
Output is correct |
31 |
Correct |
12 ms |
91228 KB |
Output is correct |
32 |
Correct |
14 ms |
91740 KB |
Output is correct |
33 |
Correct |
12 ms |
91292 KB |
Output is correct |
34 |
Correct |
14 ms |
91436 KB |
Output is correct |
35 |
Correct |
14 ms |
91484 KB |
Output is correct |
36 |
Correct |
14 ms |
91484 KB |
Output is correct |
37 |
Correct |
14 ms |
91512 KB |
Output is correct |
38 |
Correct |
12 ms |
91228 KB |
Output is correct |
39 |
Correct |
15 ms |
91456 KB |
Output is correct |
40 |
Correct |
14 ms |
91480 KB |
Output is correct |
41 |
Correct |
14 ms |
91484 KB |
Output is correct |
42 |
Correct |
14 ms |
91516 KB |
Output is correct |
43 |
Correct |
14 ms |
91740 KB |
Output is correct |
44 |
Correct |
13 ms |
91224 KB |
Output is correct |
45 |
Correct |
14 ms |
91480 KB |
Output is correct |
46 |
Correct |
15 ms |
91480 KB |
Output is correct |
47 |
Correct |
13 ms |
91244 KB |
Output is correct |
48 |
Correct |
14 ms |
91484 KB |
Output is correct |
49 |
Correct |
14 ms |
91544 KB |
Output is correct |
50 |
Correct |
14 ms |
91484 KB |
Output is correct |
51 |
Correct |
14 ms |
91736 KB |
Output is correct |
52 |
Correct |
14 ms |
91484 KB |
Output is correct |
53 |
Correct |
15 ms |
91480 KB |
Output is correct |
54 |
Correct |
17 ms |
91480 KB |
Output is correct |
55 |
Correct |
12 ms |
91228 KB |
Output is correct |
56 |
Correct |
260 ms |
124668 KB |
Output is correct |
57 |
Correct |
265 ms |
124496 KB |
Output is correct |
58 |
Correct |
14 ms |
91568 KB |
Output is correct |
59 |
Correct |
12 ms |
91228 KB |
Output is correct |
60 |
Correct |
12 ms |
91332 KB |
Output is correct |
61 |
Correct |
189 ms |
124432 KB |
Output is correct |
62 |
Correct |
14 ms |
91484 KB |
Output is correct |
63 |
Correct |
93 ms |
130896 KB |
Output is correct |
64 |
Correct |
254 ms |
124336 KB |
Output is correct |
65 |
Correct |
14 ms |
91484 KB |
Output is correct |
66 |
Correct |
102 ms |
124464 KB |
Output is correct |
67 |
Correct |
14 ms |
91484 KB |
Output is correct |
68 |
Correct |
14 ms |
91520 KB |
Output is correct |
69 |
Correct |
263 ms |
124996 KB |
Output is correct |
70 |
Correct |
14 ms |
91740 KB |
Output is correct |
71 |
Correct |
197 ms |
124060 KB |
Output is correct |
72 |
Correct |
93 ms |
126036 KB |
Output is correct |
73 |
Correct |
12 ms |
91224 KB |
Output is correct |
74 |
Correct |
113 ms |
131052 KB |
Output is correct |
75 |
Correct |
235 ms |
128852 KB |
Output is correct |
76 |
Correct |
15 ms |
91484 KB |
Output is correct |
77 |
Correct |
14 ms |
91428 KB |
Output is correct |
78 |
Correct |
264 ms |
124872 KB |
Output is correct |
79 |
Correct |
76 ms |
126932 KB |
Output is correct |
80 |
Correct |
14 ms |
91484 KB |
Output is correct |
81 |
Correct |
13 ms |
91736 KB |
Output is correct |
82 |
Correct |
12 ms |
91228 KB |
Output is correct |
83 |
Correct |
12 ms |
91216 KB |
Output is correct |
84 |
Correct |
14 ms |
91484 KB |
Output is correct |
85 |
Correct |
237 ms |
125512 KB |
Output is correct |
86 |
Correct |
14 ms |
91484 KB |
Output is correct |
87 |
Correct |
180 ms |
124612 KB |
Output is correct |
88 |
Correct |
12 ms |
91228 KB |
Output is correct |
89 |
Correct |
92 ms |
124076 KB |
Output is correct |
90 |
Correct |
192 ms |
125396 KB |
Output is correct |
91 |
Correct |
91 ms |
126800 KB |
Output is correct |
92 |
Correct |
263 ms |
124580 KB |
Output is correct |
93 |
Correct |
14 ms |
91480 KB |
Output is correct |
94 |
Correct |
12 ms |
91228 KB |
Output is correct |
95 |
Correct |
128 ms |
126472 KB |
Output is correct |
96 |
Correct |
99 ms |
130212 KB |
Output is correct |
97 |
Correct |
14 ms |
91484 KB |
Output is correct |
98 |
Correct |
12 ms |
91228 KB |
Output is correct |
99 |
Correct |
232 ms |
125376 KB |
Output is correct |
100 |
Correct |
14 ms |
91480 KB |
Output is correct |
101 |
Correct |
112 ms |
124832 KB |
Output is correct |
102 |
Correct |
110 ms |
129876 KB |
Output is correct |
103 |
Correct |
14 ms |
91484 KB |
Output is correct |
104 |
Correct |
14 ms |
91484 KB |
Output is correct |
105 |
Correct |
14 ms |
91444 KB |
Output is correct |
106 |
Correct |
14 ms |
91484 KB |
Output is correct |
107 |
Correct |
15 ms |
91484 KB |
Output is correct |
108 |
Correct |
1780 ms |
190952 KB |
Output is correct |
109 |
Correct |
796 ms |
198868 KB |
Output is correct |
110 |
Correct |
793 ms |
210140 KB |
Output is correct |
111 |
Correct |
904 ms |
217816 KB |
Output is correct |
112 |
Correct |
897 ms |
212068 KB |
Output is correct |
113 |
Correct |
778 ms |
196624 KB |
Output is correct |
114 |
Correct |
1342 ms |
183600 KB |
Output is correct |
115 |
Correct |
1332 ms |
183224 KB |
Output is correct |
116 |
Correct |
2062 ms |
188592 KB |
Output is correct |
117 |
Correct |
610 ms |
197132 KB |
Output is correct |
118 |
Correct |
1313 ms |
186072 KB |
Output is correct |
119 |
Correct |
613 ms |
197128 KB |
Output is correct |
120 |
Correct |
892 ms |
212816 KB |
Output is correct |
121 |
Correct |
594 ms |
197200 KB |
Output is correct |
122 |
Correct |
1602 ms |
192020 KB |
Output is correct |
123 |
Correct |
438 ms |
198248 KB |
Output is correct |
124 |
Correct |
644 ms |
193868 KB |
Output is correct |