QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#864359#9733. Heavy-light DecompositionJlyfishWA 6ms4096kbC++202.2kb2025-01-20 15:18:302025-01-20 15:18:30

Judging History

你现在查看的是最新测评结果

  • [2025-01-20 15:18:30]
  • 评测
  • 测评结果:WA
  • 用时:6ms
  • 内存:4096kb
  • [2025-01-20 15:18:30]
  • 提交

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;
}

詳細信息

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)