forked from daiwb/Algorithm
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path134a.cpp
More file actions
87 lines (84 loc) · 1.85 KB
/
134a.cpp
File metadata and controls
87 lines (84 loc) · 1.85 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
#include <iostream>
#include <cstdio>
#include <vector>
#include <deque>
#include <string>
#include <algorithm>
using namespace std;
#define inf 1000000000
int main() {
int n;
scanf("%d", &n);
//vector<int> cnt(n, 1);;
vector<vector<int> > link(n);
vector<int> out;
deque<int> leaf;
//vector<int> value(n, 0);
int cnt[16000], value[16000];
for (int i = 0; i < n; ++i) {
cnt[i] = 1, value[i] = 0;
}
//for (int i = 0; i < n; ++i) cout << cnt[i] << ' ' << value[i] << endl;
int k = inf;
for (int i = 0; i < n - 1; ++i) {
int a, b;
scanf("%d %d", &a, &b);
--a, --b;
link[a].push_back(b);
link[b].push_back(a);
}
if (n == 1) {
printf("0 1\n1\n");
return 0;
} else if (n == 2) {
printf("1 2\n1 2\n");
return 0;
}
for (int i = 0; i < n; ++i) {
if (link[i].size() == 1) {
leaf.push_back(i);
k = n - 1;
out.push_back(i);
}
}
while (1) {
int size = leaf.size();
if (size == 0) break;
/*
cout << "start" << endl;
for (int i = 0; i < size; ++i) {
cout << leaf[i] << ' ' << cnt[leaf[i]] << ' ' << endl;
}
cout << "end" << endl;
*/
for (int i = 0; i < size; ++i) {
int now = leaf[i];
int tmp = value[now];
if (n != cnt[now] && n - cnt[now] > tmp) tmp = n - cnt[now];
if (tmp < k) {
k = tmp;
out.clear();
out.push_back(now);
} else if (tmp == k) {
out.push_back(now);
}
int pa = link[now][0];
cnt[pa] += cnt[now];
if (cnt[now] > value[pa]) value[pa] = cnt[now];
link[pa].erase(remove(link[pa].begin(), link[pa].end(), now), link[pa].end());
if (link[pa].size() == 1) {
leaf.push_back(pa);
}
link[now].clear();
}
for (int i = 0; i < size; ++i) leaf.pop_front();
}
sort(out.begin(), out.end());
printf("%d %d\n", k, out.size());
for (int i = 0; i < out.size(); ++i) {
if (i) printf(" ");
printf("%d", out[i] + 1);
}
printf("\n");
return 0;
}