QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#252606 | #5088. Two Choreographies | NYCU_gAwr_gurA# | WA | 86ms | 26328kb | C++17 | 4.1kb | 2023-11-15 22:22:27 | 2023-11-15 22:22:27 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#ifdef DEBUG
#define fast
#else
#define fast cin.tie(0)->sync_with_stdio(0)
#define cerr if(1);else cerr
#define endl '\n'
#endif
#define _ <<' '<<
#define ALL(v) v.begin(),v.end()
#define ft first
#define sd second
using ll = long long;
using ld = long double;
using pii = pair<int,int>;
signed main() {
fast;
int n;
cin >> n;
if (n == 3) {
cout << -1 << endl;
return 0;
}
int m = 2*n-3;
vector<pii> edge(m);
vector<vector<int>> G(n);
for (auto &[u,v]: edge) {
cin >> u >> v;
u--; v--;
G[u].emplace_back(v);
G[v].emplace_back(u);
}
vector<int> par(n, -1), dep(n);
vector<pii> nt{};
int k = n-2;
nt.reserve(k);
function<void(int,int, int)> dfs;
dfs = [&](int x, int p, int d) {
par[x] = p;
dep[x] = d;
for (auto y: G[x]) if (y != p) {
if (par[y] == -1) {
dfs(y, x, d+1);
} else if (x < y) {
nt.emplace_back(x, y);
}
}
};
for (int i = 0; i < n; i++)
if (par[i] == -1)
dfs(i, i, 0);
int lgN = __lg(n)+1;
vector<vector<int>> pars(n,vector<int>(lgN));
for (int i = 0; i < n; i++)
pars[i][0] = par[i];
for (int d = 1; d < lgN; d++)
for (int i = 0; i < n; i++)
pars[i][d] = pars[pars[i][d-1]][d-1];
auto gopar = [&](int x, int w) {
for (int d = lgN-1; d >= 0; d--)
if (w >= (1<<d)) x = pars[x][d], w -= (1<<d);
return x;
};
auto lca = [&](int u, int v) {
if (dep[u] < dep[v]) swap(u, v);
u = gopar(u, dep[u] - dep[v]);
if (u == v) return u;
for (int d = lgN-1; d >= 0; d--)
if (pars[u][d] != pars[v][d])
u = pars[u][d], v = pars[v][d];
return pars[u][0];
};
auto dist = [&](int u, int v) {
auto x = lca(u, v);
return dep[u] + dep[v] - dep[x] * 2;
};
vector<vector<int>> ans{};
map<int,vector<int>> mp{};
for (int i = 0; i < k; i++) {
auto [u,v] = nt[i];
auto d = dist(u, v);
mp[d].emplace_back(i);
}
auto calc_cycle = [&](int u, int v) {
vector<int> l{}, r{};
auto x = lca(u, v);
for (int t = u; t != x; t = par[t])
l.emplace_back(t);
for (int t = v; t != x; t = par[t])
r.emplace_back(t);
reverse(ALL(r));
l.emplace_back(x);
for (auto y: r)
l.emplace_back(y);
return l;
};
for (const auto& [d,v]: mp) if (v.size() >= 2) {
for (int i = 0; i < 2; i++) {
ans.emplace_back( calc_cycle(nt[v[i]].ft, nt[v[i]].sd) );
}
break;
}
if (ans.empty()) {
vector<int> par2(n, -1), flag(n, -1);
set<pii> found, snt(ALL(nt));
function<bool(int,int)> dfs2;
dfs2 = [&](int x, int p) {
cerr _ x;
par2[x] = p;
flag[x] = flag[p];
for (auto y: G[x]) if (y != p) {
if (par2[y] == -1) {
if (snt.count({ min(x,y), max(x,y) })) continue;
if (dfs2(y, x)) return true;
} else if (flag[y] != flag[x]) {
pii p{ min(x,y), max(x,y) };
if (snt.count(p)) {
found.emplace(p);
if (found.size() >= 2)
return true;
}
}
}
return false;
};
auto calc_cycle2 = [&](int u, int v) {
vector<int> l{ u }, r{ v };
while (l.back() != par2[l.back()])
l.emplace_back(par2[l.back()]);
while (r.back() != par2[r.back()])
r.emplace_back(par2[r.back()]);
reverse(ALL(r));
for (auto y: r)
l.emplace_back(y);
return l;
};
for (auto [u,v]: nt) {
cerr _ "uv" _ u _ v _ endl;
par2.assign(n, -1);
flag.assign(n, -1);
par2[u] = u; flag[u] = 1;
par2[v] = v; flag[v] = 2;
found.clear();
cerr _ "tree u:";
dfs2(u, u);
cerr _ endl;
cerr _ "tree v:";
auto r = dfs2(v, v);
cerr _ endl;
cerr _ "found"; for (auto [x,y]: found) cerr _ x _ y _ '/'; cerr _ endl;
if (r) {
ans.emplace_back( calc_cycle(u, v) );
found.erase({ min(u, v), max(u, v) });
auto [x,y] = *found.begin();
cerr _ x _ y _ endl;
ans.emplace_back( calc_cycle2(x, y) );
break;
}
}
}
if (ans.empty()) {
cout << -1 << endl;
} else {
auto l = ans[0].size();
cout << l << endl;
for (auto& v: ans) {
for (int i = 0; i < l; i++)
cout << v[i]+1 << " \n"[i+1==l];
}
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3552kb
input:
4 1 2 1 3 1 4 2 3 2 4
output:
3 1 2 3 1 2 4
result:
ok
Test #2:
score: 0
Accepted
time: 0ms
memory: 3572kb
input:
5 1 2 1 3 1 4 1 5 2 3 2 5 3 4
output:
3 1 2 3 1 2 5
result:
ok
Test #3:
score: 0
Accepted
time: 0ms
memory: 3540kb
input:
7 1 2 3 4 5 6 5 2 3 1 6 1 4 2 4 5 2 6 3 6 1 5
output:
4 4 3 6 5 1 2 5 6
result:
ok
Test #4:
score: 0
Accepted
time: 0ms
memory: 3556kb
input:
40 1 16 1 40 2 4 2 16 2 36 3 25 3 38 4 1 4 13 5 11 5 27 6 4 7 5 7 11 8 10 8 14 8 24 9 34 10 20 12 35 13 2 14 10 14 20 15 18 15 28 15 31 16 6 16 13 17 5 17 11 17 27 18 9 19 1 19 4 19 16 20 24 21 12 21 33 21 35 22 38 23 12 23 21 25 28 25 31 25 34 25 38 26 14 26 20 27 7 27 11 28 3 28 31 29 16 29 19 30 ...
output:
3 2 4 13 1 16 40
result:
ok
Test #5:
score: 0
Accepted
time: 0ms
memory: 3804kb
input:
201 1 7 1 114 1 119 2 49 2 93 4 197 5 139 6 1 6 27 7 39 7 121 8 127 9 130 9 145 11 106 11 136 11 193 12 2 12 116 13 55 13 69 13 105 13 187 13 196 14 144 14 177 15 127 15 134 15 145 15 155 15 184 15 199 16 96 16 177 17 20 21 100 22 68 22 71 22 81 22 142 23 148 23 190 24 12 24 81 24 89 25 158 25 159 2...
output:
3 11 136 193 48 109 119
result:
ok
Test #6:
score: 0
Accepted
time: 5ms
memory: 4908kb
input:
8000 2 1508 2 3068 3 5268 3 5501 6 266 6 2737 6 3197 6 5863 6 6697 7 3492 9 427 9 794 9 3114 9 5509 10 2257 10 4348 11 1479 11 1957 11 2230 11 2500 11 3182 11 4399 11 5051 11 7727 12 7669 13 1403 13 5753 14 2871 14 6956 14 7959 15 6902 17 1630 17 3155 17 5950 18 7232 19 125 19 3280 19 5648 20 6879 2...
output:
3 2842 820 6110 3258 2997 6000
result:
ok
Test #7:
score: 0
Accepted
time: 86ms
memory: 26328kb
input:
99999 1 11261 1 21544 2 9017 2 63063 2 97990 3 11995 3 42473 4 19846 5 38099 6 35872 6 80509 7 73231 8 12356 9 35384 10 45091 12 86727 13 4938 13 48917 14 62383 14 89846 15 28458 15 44277 15 51725 15 84522 16 93258 17 13934 17 42238 18 19000 19 11278 19 23672 19 61502 19 78791 20 85057 20 88080 21 2...
output:
3 58209 60118 67758 21402 11345 61519
result:
ok
Test #8:
score: 0
Accepted
time: 83ms
memory: 26220kb
input:
100000 1 68531 2 97359 4 68578 4 83098 4 98443 5 8053 5 30270 5 86617 6 7074 6 12266 6 69396 7 52675 7 78316 7 90757 7 92242 8 32677 8 41353 8 41457 8 74508 9 44234 10 4973 10 38390 10 96049 11 28007 11 68620 13 3016 14 36748 15 8147 15 25110 15 28489 15 72947 15 99347 16 70760 17 12774 17 68407 17 ...
output:
3 48423 3475 97816 84361 56227 92343
result:
ok
Test #9:
score: -100
Wrong Answer
time: 0ms
memory: 3640kb
input:
7 1 2 2 3 3 4 4 5 5 6 6 7 7 5 1 4 7 3 1 6 7 1
output:
3 5 6 7 1 2 3
result:
wrong answer Wrong output - Nonexisting edge.