This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
#define pb push_back
#define mp make_pair
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define lb lower_bound
#define ub upper_bound
#define sz(v) int((v).size())
#define do_not_disturb ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'
void solve() {
int n;
cin >> n;
vector <int> graph[n];
for (int i = 0; i < n-1; i++) {
int a, b;
cin >> a >> b;
a--; b--;
graph[a].pb(b);
graph[b].pb(a);
}
ll cost[2];
cost[0] = cost[1] = 0;
vector <int> subtree(n);
vector <vector <int>> rear(2, vector <int> (n, 0));
function <int(int, int)> find_smallest = [&](int v, int p) {
vector <int> children;
subtree[v] = 1;
for (auto to : graph[v]) {
if (to == p) continue;
if (find_smallest(to, v)) {
children.pb(to);
}
subtree[v] += subtree[to];
}
if (sz(children) == 0) {
if (v) return 1;
cost[0] += 2;
int x = graph[v][0];
rear[0][v] = rear[0][x];
rear[0][x] = v;
}
else if (sz(children)&1) {
cost[0] += int(sz(children)/2)*4+2;
rear[0][v] = children[0];
rear[0][children[0]] = v;
for (int i = 1; i < sz(children); i += 2) {
rear[0][children[i]] = children[i+1];
rear[0][children[i+1]] = children[i];
}
}
else {
cost[0] += sz(children)*2;
rear[0][v] = children[1];
rear[0][children[1]] = children[0];
rear[0][children[0]] = v;
for (int i = 2; i < sz(children); i += 2) {
rear[0][children[i]] = children[i+1];
rear[0][children[i+1]] = children[i];
}
}
return 0;
};
find_smallest(0, 0);
function <int(int, int)> find_centroid = [&](int v, int p) {
for (auto to : graph[v]) {
if (to == p) continue;
if (subtree[to]*2 > n) {
return find_centroid(to, v);
}
}
return v;
};
function <void(int, int)> find_largest = [&](int v, int p) {
if (v) {
cost[1] += min(subtree[v], subtree[0]-subtree[v])*2;
}
for (auto to : graph[v]) {
if (to == p) continue;
find_largest(to, v);
}
};
int centroid = find_centroid(0, 0);
find_largest(0, 0);
vector <vector <int>> children(sz(graph[centroid]));
priority_queue <pii> st;
function <void(int, int, int)> find_children = [&](int v, int p, int i) {
children[i].pb(v);
for (auto to : graph[v]) {
if (to == p) continue;
find_children(to, v, i);
}
};
for (int i = 0; i < sz(graph[centroid]); i++) {
find_children(graph[centroid][i], centroid, i);
st.push(mp(sz(children[i]), i));
}
while (sz(st) > 1) {
auto l = st.top(); st.pop();
auto r = st.top(); st.pop();
l.first--; r.first--;
rear[1][children[l.second].back()] = children[r.second].back();
rear[1][children[r.second].back()] = children[l.second].back();
children[l.second].pop_back();
children[r.second].pop_back();
if (l.first) st.push(l);
if (r.first) st.push(r);
}
if (sz(st)) {
auto l = st.top().second;
rear[1][centroid] = children[l].back();
rear[1][children[l].back()] = centroid;
}
else {
auto x = graph[centroid][0];
rear[1][centroid] = rear[1][x];
rear[1][x] = centroid;
}
cout << cost[0] << ' ' << cost[1] << endl;
for (int i = 0; i < 2; i++) {
for (auto to : rear[i]) cout << to+1 << ' ';
cout << endl;
}
}
int main() {
//~ do_not_disturb
int t = 1;
//~ cin >> t;
while (t--) {
solve();
}
return 0;
}
/*
7
4 2
5 7
3 4
6 3
1 3
4 5
*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |