-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathccc16s3.cpp
46 lines (46 loc) · 1.13 KB
/
ccc16s3.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
// Ivan Carvalho
// Solution to https://dmoj.ca/problem/ccc16s3
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;
const int MAXN = 100010;
vector<int> grafo[MAXN];
int marcado[MAXN], valido[MAXN], aux, diametro, primeiro, n, m, novo;
void dfs(int v, int p) {
valido[v] = marcado[v];
for (int u : grafo[v]) {
if (u == p) continue;
dfs(u, v);
valido[v] |= valido[u];
}
if (valido[v]) novo++;
}
void find_diametro(int v, int p, int depth) {
if (depth > diametro) {
diametro = depth;
aux = v;
}
for (int u : grafo[v]) {
if (u == p || !valido[u]) continue;
find_diametro(u, v, depth + 1);
}
}
int main() {
scanf("%d %d", &n, &m);
for (int i = 1; i <= m; i++) {
scanf("%d", &primeiro);
marcado[primeiro] = 1;
}
for (int i = 1; i < n; i++) {
int u, v;
scanf("%d %d", &u, &v);
grafo[u].push_back(v);
grafo[v].push_back(u);
}
dfs(primeiro, -1);
find_diametro(primeiro, -1, 0);
find_diametro(aux, -1, 0);
printf("%d\n", 2 * (novo - 1) - diametro);
return 0;
}