QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#445169#8529. Balance of Permutationucup-team3215#ML 5735ms930532kbC++203.1kb2024-06-15 23:54:392024-06-15 23:54:40

Judging History

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

  • [2024-06-15 23:54:40]
  • 评测
  • 测评结果:ML
  • 用时:5735ms
  • 内存:930532kb
  • [2024-06-15 23:54:39]
  • 提交

answer

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#pragma GCC target "abm,bmi,bmi2"

using namespace std;

constexpr int N = 1e6;

using u32 = uint32_t;
using u64 = uint64_t;
using u128 = __uint128_t;

u64 splitmix(u64 x) {
  x += 0x9e3779b97f4a7c15;
  (x ^= x >> 30) *= 0xbf58476d1ce4e5b9;
  (x ^= x >> 27) *= 0x94d049bb133111eb;
  return x ^ (x >> 31);
}

template <typename V>
struct umap {
  vector<u64> keys;
  vector<V> vals;
  vector<u32> nx;
  vector<u32> head;
  int cap{};

  int geti(u64 k) {
    reserve();
    k = splitmix(k);
    for (int i = head[k & cap]; ~i; i = nx[i]) if (keys[i] == k) return i;
    nx.push_back(head[k & cap]);
    head[k & cap] = keys.size();
    keys.push_back(k);
    vals.push_back({});
    return vals.size() - 1;
  }

  void reserve() {
    if (keys.size() * 2 < cap) return;
    cap = max(cap * 2 + 1, 7);
    head.assign(cap + 1, -1);
    nx.clear();
    for (int i = 0; i < keys.size(); ++i) {
      nx.push_back(head[keys[i] & cap]);
      head[keys[i] & cap] = i;
    }
  }
};

int n, b;

umap<u64> mem_[30][30][256];

u64 cnt_(int l, int c, int b, u32 m) {
  if (m >> l & 1) ++c, m -= 1 << l;
  if (l == n || b < 0) return !b;
  int loc = mem_[l][c][b].geti(m);
  if (mem_[l][c][b].vals[loc]) return ~mem_[l][c][b].vals[loc];
  int mx = (2 * n - 1 - c - 2 * l) * c / 2;
  for (int i = n - c, t = m; --i > l; ) {
    if (int d = __builtin_ctz(t); d < i) mx += i - d, t &= t - 1;
    else break;
  }
  if (b < 0 || b > mx) return mem_[l][c][b].vals[loc] = -1, 0;
  u64 ans{};
  if (c) ans += cnt_(l + 1, c - 1, b - c + 1, m) * c;
  for (u32 t = m; t; t &= t - 1) ans += cnt_(l + 1, c, b - c, m - (t & -t));
  mem_[l][c][b].vals[loc] = ~ans;
  return ans;
}

umap<u128> mem[30][30][256];

u128 cnt(int l, int c, int b, u32 m) {
  if (n - l < 21) return cnt_(l, c, b, m);
  if (m >> l & 1) ++c, m -= 1 << l;
  if (l == n || b < 0) return !b;
  int loc = mem[l][c][b].geti(m);
  if (mem[l][c][b].vals[loc]) return ~mem[l][c][b].vals[loc];
  int mx = (2 * n - 1 - c - 2 * l) * c / 2;
  for (int i = n - c, t = m; --i > l; ) {
    if (int d = __builtin_ctz(t); d < i) mx += i - d, t &= t - 1;
    else break;
  }
  if (b < 0 || b > mx) return mem[l][c][b].vals[loc] = -1, 0;
  u128 ans{};
  if (c) ans += cnt(l + 1, c - 1, b - c + 1, m) * c;
  for (u32 t = m; t; t &= t - 1) ans += cnt(l + 1, c, b - c, m - (t & -t));
  mem[l][c][b].vals[loc] = ~ans;
  return ans;
}

u128 cnt(int l, int b, u32 m) {
  int c = 0;
  for (int i = 0; i < l; ++i) if (m & 1 << i) {
    ++c;
    b -= l - i;
    m ^= 1 << i;
  }
  auto ans = cnt(l, c, b, m);
  return ans;
}

int main() {
  cin.tie(0)->sync_with_stdio(0);
  cin >> n >> b; b /= 2;
  u128 k{};
  for (char c; cin >> c && isdigit(c); ) k = k * 10 + c - '0';
  u32 ans[30], l = 0, m = (1 << n) - 1;
  while (l < n)
  for (int i = 0; i < n; ++i) if (m & 1 << i) if (auto t = cnt(l + 1, b - max((int)l - i, {}), m - (1 << i)); t >= k) {
    b -= max((int)l - i, {});
    ans[l++] = i + 1;
    m -= 1 << i;
    break;
  } else k -= t;
  for (int i = 0; i < n; ++i) cout << ans[i] << ' ';
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 3740kb

input:

6 6 6

output:

1 2 6 3 4 5 

result:

ok 6 numbers

Test #2:

score: 0
Accepted
time: 3324ms
memory: 584904kb

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: 3636kb

input:

1 0 1

output:

1 

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 2ms
memory: 3664kb

input:

2 0 1

output:

1 2 

result:

ok 2 number(s): "1 2"

Test #5:

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

input:

2 2 1

output:

2 1 

result:

ok 2 number(s): "2 1"

Test #6:

score: 0
Accepted
time: 2ms
memory: 3724kb

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: 2ms
memory: 3624kb

input:

7 20 100

output:

3 6 7 4 1 5 2 

result:

ok 7 numbers

Test #8:

score: 0
Accepted
time: 2ms
memory: 3684kb

input:

7 2 6

output:

2 1 3 4 5 6 7 

result:

ok 7 numbers

Test #9:

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

input:

7 24 1

output:

4 5 6 7 1 2 3 

result:

ok 7 numbers

Test #10:

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

input:

7 22 360

output:

7 6 4 3 5 2 1 

result:

ok 7 numbers

Test #11:

score: 0
Accepted
time: 2ms
memory: 3724kb

input:

7 20 358

output:

5 7 2 4 6 3 1 

result:

ok 7 numbers

Test #12:

score: 0
Accepted
time: 2ms
memory: 3664kb

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: 2ms
memory: 3716kb

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: 3ms
memory: 4596kb

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: 20ms
memory: 11212kb

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: 159ms
memory: 54540kb

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: 519ms
memory: 131388kb

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: 1414ms
memory: 295868kb

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: 3008ms
memory: 529992kb

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: 2525ms
memory: 479912kb

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: 15ms
memory: 14924kb

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: 5609ms
memory: 930532kb

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: 5735ms
memory: 922544kb

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: 107ms
memory: 57844kb

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: 22ms
memory: 14948kb

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: 2543ms
memory: 481664kb

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: 2ms
memory: 3764kb

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: 7ms
memory: 9176kb

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: 17ms
memory: 15012kb

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: 12ms
memory: 14952kb

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: -100
Memory Limit Exceeded

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: