QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#443411#8529. Balance of Permutationucup-team133#AC ✓76ms4200kbC++233.8kb2024-06-15 15:28:512024-06-15 15:29:19

Judging History

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

  • [2024-06-15 15:29:19]
  • 评测
  • 测评结果:AC
  • 用时:76ms
  • 内存:4200kb
  • [2024-06-15 15:28:51]
  • 提交

answer

#include <bits/stdc++.h>
#ifdef LOCAL
#include <debug.hpp>
#else
#define debug(...) void(0)
#endif

template <class T> std::istream& operator>>(std::istream& is, std::vector<T>& v) {
    for (auto& e : v) {
        is >> e;
    }
    return is;
}

template <class T> std::ostream& operator<<(std::ostream& os, const std::vector<T>& v) {
    for (std::string_view sep = ""; const auto& e : v) {
        os << std::exchange(sep, " ") << e;
    }
    return os;
}

template <class T, class U = T> bool chmin(T& x, U&& y) {
    return y < x and (x = std::forward<U>(y), true);
}

template <class T, class U = T> bool chmax(T& x, U&& y) {
    return x < y and (x = std::forward<U>(y), true);
}

template <class T> void mkuni(std::vector<T>& v) {
    std::ranges::sort(v);
    auto result = std::ranges::unique(v);
    v.erase(result.begin(), result.end());
}

template <class T> int lwb(const std::vector<T>& v, const T& x) {
    return std::distance(v.begin(), std::ranges::lower_bound(v, x));
}

using ll = __int128_t;

using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout << fixed << setprecision(15);
    int n, b;
    string tmp;
    cin >> n >> b >> tmp;

    ll K = 0;
    for (char& c : tmp) K = K * 10 + (c - '0');
    K--;
    vector<int> ans;
    vector<bool> used(n, false);
    for (int p = 0; p < n; p++) {
        for (int d = 0; d < n; d++) {
            if (used[d]) continue;
            int nb = b - abs(p - d);
            if (nb < 0) continue;
            used[d] = true;
            int cnt = 0, sum = 0;
            for (int i = 0; i <= p; i++) {
                if (used[i]) continue;
                cnt++;
                sum += abs(p + 1 - i);
            }
            if (sum > nb) continue;
            nb -= sum;
            vector dp(n + 1, vector<ll>(nb + 1, 0)), ndp(n + 1, vector<ll>(nb + 1, 0));
            dp[cnt][0] = 1;
            debug(nb);
            for (int k = p + 1, add = 0; k < n; k++, add++) {
                for (int i = 0; i <= n; i++) {
                    for (int j = 0; j <= nb; j++) {
                        ll& val = dp[i][j];
                        if (val == 0) continue;
                        int x = not used[k];
                        int rest = add - (cnt - i);
                        if (i > 0 and rest > 0 and not used[k]) {  // use a and b
                            int ni = i - 1, nj = j + ni + (rest - 1);
                            if (nj <= nb) ndp[ni][nj] += val * i * rest;
                        }
                        if (not used[k]) {  // link a and b
                            int ni = i, nj = j + ni + rest;
                            if (nj <= nb) ndp[ni][nj] += val;
                        }
                        if (i > 0) {  // use a
                            int ni = i - 1 + x, nj = j + ni + rest;
                            if (nj <= nb) ndp[ni][nj] += val * i;
                        }
                        if (not used[k]) {  // use b
                            int ni = i, nj = j + ni + rest;
                            if (nj <= nb) ndp[ni][nj] += val * rest;
                        }
                        {
                            int ni = i + x, nj = j + ni + rest + 1;
                            if (nj <= nb) ndp[ni][nj] += val;
                        }
                        val = 0;
                    }
                }
                cnt += (not used[k]);
                swap(dp, ndp);
            }
            ll val = dp[0][nb];
            debug(p, d, (long long)val);
            if (K < val) {
                ans.emplace_back(d);
                break;
            } else {
                K -= val;
                used[d] = false;
            }
        }
        b -= abs(p - ans.back());
        debug(used);
    }

    for (int i = 0; i < n; i++) cout << ans[i] + 1 << (i + 1 == n ? "\n" : " ");
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3552kb

input:

6 6 6

output:

1 2 6 3 4 5

result:

ok 6 numbers

Test #2:

score: 0
Accepted
time: 23ms
memory: 4124kb

input:

30 300 3030303030303030303030

output:

1 2 3 4 9 23 20 28 24 16 21 17 27 29 8 26 25 30 19 18 22 12 7 13 6 10 5 15 14 11

result:

ok 30 numbers

Test #3:

score: 0
Accepted
time: 0ms
memory: 3548kb

input:

1 0 1

output:

1

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3608kb

input:

2 0 1

output:

1 2

result:

ok 2 number(s): "1 2"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3540kb

input:

2 2 1

output:

2 1

result:

ok 2 number(s): "2 1"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3560kb

input:

5 8 3

output:

1 5 4 2 3

result:

ok 5 number(s): "1 5 4 2 3"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3564kb

input:

7 20 100

output:

3 6 7 4 1 5 2

result:

ok 7 numbers

Test #8:

score: 0
Accepted
time: 0ms
memory: 3552kb

input:

7 2 6

output:

2 1 3 4 5 6 7

result:

ok 7 numbers

Test #9:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

7 24 1

output:

4 5 6 7 1 2 3

result:

ok 7 numbers

Test #10:

score: 0
Accepted
time: 0ms
memory: 3576kb

input:

7 22 360

output:

7 6 4 3 5 2 1

result:

ok 7 numbers

Test #11:

score: 0
Accepted
time: 0ms
memory: 3572kb

input:

7 20 358

output:

5 7 2 4 6 3 1

result:

ok 7 numbers

Test #12:

score: 0
Accepted
time: 1ms
memory: 3632kb

input:

10 48 10001

output:

7 5 8 9 6 10 3 4 1 2

result:

ok 10 numbers

Test #13:

score: 0
Accepted
time: 1ms
memory: 3680kb

input:

10 42 10101

output:

3 9 6 8 10 5 7 2 1 4

result:

ok 10 numbers

Test #14:

score: 0
Accepted
time: 18ms
memory: 4012kb

input:

25 300 1

output:

7 14 15 16 17 18 19 20 21 22 23 24 25 1 2 3 4 5 6 8 9 10 11 12 13

result:

ok 25 numbers

Test #15:

score: 0
Accepted
time: 25ms
memory: 3796kb

input:

25 300 283788388040048639877

output:

25 24 23 22 21 20 19 18 17 16 11 12 13 14 15 10 9 8 7 5 6 4 2 1 3

result:

ok 25 numbers

Test #16:

score: 0
Accepted
time: 24ms
memory: 4064kb

input:

26 302 105773752969551707419545

output:

19 22 25 13 17 18 23 20 10 26 16 6 5 11 14 12 24 4 3 21 1 15 7 8 2 9

result:

ok 26 numbers

Test #17:

score: 0
Accepted
time: 27ms
memory: 3908kb

input:

27 308 8781128321749037280676555

output:

16 18 17 21 25 6 20 24 22 15 27 5 7 8 2 9 26 13 1 3 14 10 23 19 4 11 12

result:

ok 27 numbers

Test #18:

score: 0
Accepted
time: 26ms
memory: 3884kb

input:

28 304 806517199954337651602356955

output:

12 17 5 16 23 26 25 15 20 2 19 7 22 24 6 13 11 10 28 8 1 21 18 14 27 3 4 9

result:

ok 28 numbers

Test #19:

score: 0
Accepted
time: 33ms
memory: 4080kb

input:

29 322 40281026669581503094652149519

output:

16 21 10 25 17 29 9 28 2 8 26 27 22 4 3 5 18 14 19 1 23 20 15 11 13 7 6 12 24

result:

ok 29 numbers

Test #20:

score: 0
Accepted
time: 52ms
memory: 3884kb

input:

30 400 46479902466857426153849991132

output:

25 19 30 29 9 20 26 21 14 27 28 10 22 11 24 2 7 4 18 17 5 13 12 6 8 1 15 23 16 3

result:

ok 30 numbers

Test #21:

score: 0
Accepted
time: 60ms
memory: 3952kb

input:

30 450 1140008168482799670544355

output:

26 16 17 18 19 20 21 22 23 24 25 27 28 29 30 1 2 3 5 9 4 8 14 10 6 11 12 15 7 13

result:

ok 30 numbers

Test #22:

score: 0
Accepted
time: 8ms
memory: 3700kb

input:

30 150 480087379811286955791425915

output:

7 4 8 5 16 3 1 12 13 11 9 10 15 25 18 17 20 30 28 2 6 14 23 21 24 26 27 22 19 29

result:

ok 30 numbers

Test #23:

score: 0
Accepted
time: 8ms
memory: 3656kb

input:

30 150 480087379811286955791439470

output:

7 4 8 5 16 3 1 12 13 11 9 10 15 25 18 17 20 30 28 2 19 6 22 24 21 23 26 14 29 27

result:

ok 30 numbers

Test #24:

score: 0
Accepted
time: 58ms
memory: 3868kb

input:

30 440 41509275104334759322587324

output:

22 23 20 24 18 30 19 26 21 28 4 29 17 25 27 16 3 1 2 5 8 13 10 15 7 12 9 14 11 6

result:

ok 30 numbers

Test #25:

score: 0
Accepted
time: 55ms
memory: 3896kb

input:

30 450 1140008168482800727111311

output:

26 16 17 18 19 20 21 22 23 24 25 27 28 29 30 1 2 5 7 14 4 15 8 11 3 13 10 9 6 12

result:

ok 30 numbers

Test #26:

score: 0
Accepted
time: 58ms
memory: 3876kb

input:

30 400 52289890275214604423031772929

output:

26 27 29 21 28 16 18 11 2 25 24 23 6 30 20 13 17 10 15 4 9 12 8 22 19 1 5 7 3 14

result:

ok 30 numbers

Test #27:

score: 0
Accepted
time: 0ms
memory: 3508kb

input:

30 0 1

output:

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

result:

ok 30 numbers

Test #28:

score: 0
Accepted
time: 54ms
memory: 3844kb

input:

30 450 1

output:

16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15

result:

ok 30 numbers

Test #29:

score: 0
Accepted
time: 76ms
memory: 4196kb

input:

30 450 1710012252724199424000000

output:

30 29 28 27 26 25 24 23 22 21 20 19 18 17 16 15 14 13 12 11 10 9 8 7 6 5 4 3 2 1

result:

ok 30 numbers

Test #30:

score: 0
Accepted
time: 72ms
memory: 3844kb

input:

30 450 1692383260428073656742269

output:

30 27 26 28 18 29 21 19 25 17 20 16 24 22 23 7 13 4 6 3 5 12 1 15 14 9 11 8 2 10

result:

ok 30 numbers

Test #31:

score: 0
Accepted
time: 39ms
memory: 4080kb

input:

30 302 5918364042599361729860937331200

output:

30 29 28 27 26 25 14 8 9 10 11 12 13 7 15 16 17 18 19 20 21 22 23 24 6 5 4 3 2 1

result:

ok 30 numbers

Test #32:

score: 0
Accepted
time: 23ms
memory: 3744kb

input:

30 254 2256781660157136563723839089600

output:

25 2 3 12 7 16 19 8 22 6 11 17 27 26 10 24 15 21 20 18 28 9 30 23 14 13 5 29 4 1

result:

ok 30 numbers

Test #33:

score: 0
Accepted
time: 63ms
memory: 4200kb

input:

30 448 3131906441000512625049600

output:

23 20 28 18 26 30 19 29 27 22 17 24 21 25 2 13 16 15 14 12 11 10 9 8 7 6 5 4 3 1

result:

ok 30 numbers

Test #34:

score: 0
Accepted
time: 1ms
memory: 3556kb

input:

30 2 20

output:

1 2 3 4 5 6 7 8 9 11 10 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30

result:

ok 30 numbers

Test #35:

score: 0
Accepted
time: 0ms
memory: 3548kb

input:

30 2 29

output:

2 1 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

result:

ok 30 numbers

Extra Test:

score: 0
Extra Test Passed