-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathC.cpp
112 lines (96 loc) · 2.45 KB
/
C.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
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include <bits/stdc++.h>
using namespace std;
//----------------------------------- DEBUG -----------------------------------
#define sim template < class c
#define ris return * this
#define dor > debug & operator <<
#define eni(x) sim > typename \
enable_if<sizeof dud<c>(0) x 1, debug&>::type operator<<(c i) {
sim > struct rge { c b, e; };
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c* x) -> decltype(cerr << *x, 0);
sim > char dud(...);
struct debug {
#ifdef LOCAL
~debug() { cerr << endl; }
eni(!=) cerr << boolalpha << i; ris; }
eni(==) ris << range(begin(i), end(i)); }
sim, class b dor(pair < b, c > d) {
ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
*this << "[";
for (auto it = d.b; it != d.e; ++it)
*this << ", " + 2 * (it == d.b) << *it;
ris << "]";
}
#else
sim dor(const c&) { ris; }
#endif
};
#define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
// debug & operator << (debug & dd, P p) { dd << "(" << p.x << ", " << p.y << ")"; return dd; }
//----------------------------------- END DEBUG --------------------------------
int main() {
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int t;
cin >> t;
while(t--) {
int n;
cin >> n;
vector<vector<int>> adj(n);
for(int i=0;i<n-1;i++) {
int u,v;
cin >> u >> v;
--u; --v;
adj[u].push_back(v);
adj[v].push_back(u);
}
vector<int> centroids;
vector<int> sz(n);
int leaf,p;
function<void(int,int)> dfs = [&](int node, int par) {
sz[node] = 1;
bool is_centroid = true;
for(auto u: adj[node]) {
if(u != par) {
dfs(u, node);
if(sz[u] > n / 2) {
is_centroid = false;
}
sz[node] += sz[u];
}
}
if(n - sz[node] > n / 2) {
is_centroid = false;
}
if(is_centroid) {
// debug() << imie(sz[node]);
centroids.push_back(node);
}
};
function<void(int,int)> findLeaf = [&](int node, int par) {
for(auto u: adj[node]) {
if(u != par) {
if((int)adj[u].size() == 1) {
p = node;
leaf = u;
return;
}
findLeaf(u, node);
}
}
};
dfs(0,-1);
if((int)centroids.size() == 1) {
cout << centroids[0] + 1 << " " << adj[centroids[0]][0] + 1 << '\n';
cout << centroids[0] + 1 << " " << adj[centroids[0]][0] + 1<< '\n';
}
// debug() << imie(centroids);
else {
findLeaf(centroids[0],centroids[1]);
cout << p + 1 << " " << leaf + 1 << '\n';
cout << centroids[1] + 1 << " " << leaf + 1 << '\n';
}
}
}