-
Notifications
You must be signed in to change notification settings - Fork 18
Expand file tree
/
Copy pathB.cpp
More file actions
92 lines (77 loc) · 2.07 KB
/
B.cpp
File metadata and controls
92 lines (77 loc) · 2.07 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <bits/stdc++.h>
#define int long long
#define FOR(i,a,b) for(int i=(a),_b=(b); i<=_b; ++i)
#define FORD(i,a,b) for(int i=(a),_b=(b); i>=_b; --i)
#define REP(i,a) for(int i=0,_a=(a); i < _a; ++i)
#define DEBUG(X) { cout << #X << " = " << (X) << endl; }
#define PR(A,n) { cout << #A << " = "; FOR(_,1,n) cout << A[_] << ' '; cout << endl; }
#define PR0(A,n) { cout << #A << " = "; REP(_,n) cout << A[_] << ' '; cout << endl; }
#define sqr(x) ((x) * (x))
#define ll long long
#define __builtin_popcount __builtin_popcountll
#define SZ(x) ((int) (x).size())
using namespace std;
double safe_sqrt(double x) {
return sqrt(max((double)0.0,x));
}
int GI(ll& x) {
return scanf("%lld", &x);
}
const int MN = 200111;
int n, k;
vector<int> ke[MN];
int sz[MN];
int good[MN], f[MN];
int center = 0;
void dfs1(int u, int fu) {
sz[u] = good[u];
for(int v : ke[u]) {
if (v == fu) continue;
dfs1(v, u);
sz[u] += sz[v];
}
int ln = k - sz[u];
if (good[u]) ln = max(ln, 1LL);
for(int v : ke[u]) {
if (v == fu) continue;
ln = max(ln, sz[v]);
}
if (ln * 2 <= k) {
center = u;
}
}
void dfs2(int u, int fu) {
for(int v : ke[u]) {
if (v == fu) continue;
f[v] = f[u] + 1;
dfs2(v, u);
}
}
#undef int
int main() {
#define int long long
ios :: sync_with_stdio(0); cin.tie(0);
cout << (fixed) << setprecision(9);
while (GI(n) == 1 && GI(k) == 1) {
k *= 2;
FOR(i,1,n) ke[i].clear();
memset(sz, 0, sizeof sz);
memset(good, 0, sizeof good);
FOR(i,1,k) {
int u; GI(u); good[u] = 1;
}
FOR(i,2,n) {
int u, v; GI(u); GI(v);
ke[u].push_back(v);
ke[v].push_back(u);
}
center = -1;
dfs1(1, -1);
if (center < 0) throw 1;
f[center] = 0;
dfs2(center, -1);
int res = 0;
FOR(i,1,n) if (good[i]) res += f[i];
cout << res << endl;
}
}