QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#766844 | #9733. Heavy-light Decomposition | i_wish_a_girl_friend# | WA | 3ms | 3888kb | C++20 | 2.0kb | 2024-11-20 18:51:45 | 2024-11-20 18:51:52 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
// #define max(a, b) ((a) > (b) ? (a) : (b))
// #define min(a, b) ((a) < (b) ? (a) : (b))
#define pb push_back
#define LNF 1e18
#define INF 0x7fffffff
#define int long long
#define lowbit(x) ((x) & (-x))
#define abs(x) llabs(x)
#define endl '\n'
#define Y() cout << "Yes" << endl
#define N() cout << "No" << endl
const db eps = 1e-9;
const int mod = 998244353;
const int MAXN = 2e5 + 5;
void solve()
{
int n, k;
cin >> n >> k;
vector<pair<int, int>> v;
for (int i = 1; i <= k; i++)
{
int a, b;
cin >> a >> b;
v.push_back({a, b});
}
sort(v.begin(), v.end(), [&](const pair<int, int> &a, const pair<int, int> &b)
{ return a.second - a.first > b.second - b.first; });
vector<int> fa(n + 1);
auto [l1, r1] = v[0];
int rt = l1;
int len = r1 - l1;
if (len == 0)
{
if (n != 1)
{
cout << "IMPOSSIBLE" << endl;
return;
}
else
{
cout << 0 << endl;
return;
}
}
int num = len, ma = 0;
for (int i = 1; i < v.size(); i++)
{
auto [a, b] = v[i];
if (b - a + 1 >= len)
{
fa[a] = rt;
ma = max(ma, b - a + 1);
}
else
{
num += b - a + 1;
fa[a] = rt + 1;
}
}
if (ma > len+num)
{
cout << "IMPOSSIBLE" << endl;
}
else
{
for (auto [a, b] : v)
{
for (int i = a + 1; i <= b; i++)
{
fa[i] = i - 1;
}
}
for (int i = 1; i <= n; i++)
cout << fa[i] << " ";
cout << endl;
}
}
signed main()
{
std::ios::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
cin >> T;
while (T--)
solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3660kb
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 2 2 7 2 9 10 2 2 0 2 2 IMPOSSIBLE
result:
ok Correct. (3 test cases)
Test #2:
score: -100
Wrong Answer
time: 3ms
memory: 3888kb
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 2 should be a heavy child, but the maximum one with size 3 and 2 found. (test case 3)