QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#864359 | #9733. Heavy-light Decomposition | Jlyfish | WA | 6ms | 4096kb | C++20 | 2.2kb | 2025-01-20 15:18:30 | 2025-01-20 15:18:30 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
void solve() {
int n, k;
cin >> n >> k;
vector<pair<int, int>> a(k);
for (auto &[l, r] : a) {
cin >> l >> r;
}
if (n == 1) {
cout << "0\n";
return;
}
vector<int> sum(n + 2);
for (auto &[l, r] : a) {
sum[l]++;
sum[r + 1]--;
}
for (int i = 1; i <= n; i++) {
sum[i] += sum[i - 1];
if (sum[i] != 1) {
cout << "IMPOSSIBLE\n";
return;
}
}
ranges::sort(a, [&](auto l, auto r) {
return l.second - l.first > r.second - r.first;
});
int root = a[0].first;
if (k != 1 && a.back().second - a.back().first == a.front().second - a.front().first) {
cout << "IMPOSSIBLE\n";
return;
}
if (a.front().second - a.front().first == 1 && a[1].second - a[1].first == 1) {
cout << "IMPOSSIBLE\n";
return;
}
vector<int> fa(n + 1);
if (k != 1 && a.front().second - a.front().first == a[1].second - a[1].first) {
if (a.back().second - a.back().first - 1 == a.front().second - a.front().first) {
cout << "IMPOSSIBLE\n";
return;
}
for (int i = a[0].first + 1; i <= a[0].second; i++) {
fa[i] = i - 1;
}
assert(a[0].first + 1 <= a[0].second);
fa[a.back().first] = a[0].first + 1;
for (int i = a.back().first + 1; i <= a.back().second; i++) {
fa[i] = i - 1;
}
a.erase(a.begin());
a.erase(prev(a.end()));
for (auto &[l, r] : a) {
fa[l] = root;
for (int i = l + 1; i <= r; i++) {
fa[i] = i - 1;
}
}
} else {
for (auto &[l, r] : a) {
if (l != root) {
fa[l] = root;
}
for (int i = l + 1; i <= r; i++) {
fa[i] = i - 1;
}
}
}
for (int i = 1; i <= n; i++) {
cout << fa[i] << " \n"[i == n];
}
}
int main() {
cin.tie(nullptr)->sync_with_stdio(false);
int TT;
cin >> TT;
while (TT--) {
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3712kb
input:
3 12 5 1 5 9 11 7 8 6 6 12 12 4 3 1 1 4 4 2 3 2 2 1 1 2 2
output:
0 1 2 3 4 1 1 7 1 9 10 1 2 0 2 2 IMPOSSIBLE
result:
ok Correct. (3 test cases)
Test #2:
score: -100
Wrong Answer
time: 6ms
memory: 4096kb
input:
10 1 1 1 1 100000 1 1 100000 12 4 1 3 4 6 7 9 10 12 6 3 4 6 2 3 1 1 8999 3 1 3000 3001 6000 6001 8999 14 4 1 3 4 6 7 10 11 14 17 5 1 3 4 6 7 10 11 14 15 17 19999 2 1 9999 10000 19999 1 1 1 1 5 3 1 1 2 3 4 5
output:
0 0 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 ...
result:
wrong answer 3 should be a heavy child, but the maximum one with size 2999 and 2998 found. (test case 5)