-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwc_2018_c3_s2.cpp
58 lines (53 loc) · 1.14 KB
/
wc_2018_c3_s2.cpp
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
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pi;
typedef long long ll;
typedef vector<int> vi;
#define scan(x) do{while((x=getchar())<'0'); for(x-='0'; '0'<=(_=getchar()); x=(x<<3)+(x<<1)+_-'0');}while(0)
char _;
int N,K,F,len,far,cntG;
vi tree[100001];
bool G[100001];
void dfsG(int u, int par) {
for(int v: tree[u]) {
if(v==par) continue;
dfsG(v,u);
G[u] |= G[v];
}
}
void dfsD(int u, int par, int dis) {
if(dis > len) {
len = dis;
far = u;
}
for(int v: tree[u]) {
if(v == par || !G[v]) continue;
dfsD(v,u,dis+1);
}
}
int main() {
scan(N); scan(K); scan(F);
for(int i = 0,inp; i < K; i++) {
scan(inp);
G[inp] = true;
}
for(int i = 1,u,v; i < N; i++) {
cin >> u >> v;
tree[u].push_back(v);
tree[v].push_back(u);
}
// cout << 1;
dfsG(1,-1);
dfsD(1, -1, 0);
for(bool i: G) {
cntG+=i;
}
int normalTime = len + (cntG-1-len) * 2;
cntG=0;
G[F] = true;
dfsG(1,-1);
for(bool i: G) {
cntG+=i;
}
cout << min(cntG-1,normalTime) << "\n";
}