QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#405538#7613. Inverse Problemucup-team3215WA 118ms3700kbC++204.9kb2024-05-06 06:26:212024-05-06 06:26:21

Judging History

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

  • [2024-05-06 06:26:21]
  • 评测
  • 测评结果:WA
  • 用时:118ms
  • 内存:3700kb
  • [2024-05-06 06:26:21]
  • 提交

answer

#include <algorithm>
#include <cstdint>
#include <iostream>
#include <numeric>
#include <vector>

using namespace std;

constexpr int mod = 1e9 + 7;

auto& mul(auto&& a, auto b) { return a = a * (uint64_t) b % mod; }
int inv(int a) { for (int r = 1, b = mod - 2; ; b /= 2, mul(a, a)) if (!b) return r; else if (b % 2) mul(r, a); }

using namespace std;

template <typename K, typename V, int mlf = 4>
struct HT {
  int capacity;
  vector<pair<K, int>> knx;
  vector<V> values;
  vector<int> head;

  HT(): capacity{4 * mlf}, head(capacity, -1) {
  }

  void emplace(K k, V v) {
    knx.push_back({k, head[k & capacity - 1]});
    values.push_back(v);
    head[k & capacity - 1] = knx.size() - 1;
    maybe_rehash();
  }

  int find(K k) {
    for (int i = head[k & capacity - 1]; ~i; i = knx[i].second) if (knx[i].first == k) return i;
    return -1;
  }

  void clear() {
    knx.clear();
    values.clear();
    head.assign(capacity, -1);
  }

  void maybe_rehash() {
    if (knx.size() * mlf <= capacity) return;
    head.assign(capacity *= 2, -1);
    for (int i = 0; i < knx.size(); ++i) {
      knx[i].second = head[knx[i].first & capacity - 1];
      head[knx[i].first & capacity - 1] = i;
    }
  }
};

vector<HT<int, vector<int>>> high(1);
vector<int> inv_(1);

void gen_high(const vector<int>& prof, int n0, int n, int c, int p, auto&& cur) {
  high[n0 - n].emplace(p, cur);
  for (c = min(n, c); c > prof[n]; --c) {
    int np = p;
    for (int i = 0; i < c; ++i) mul(np, inv_[n0 - i]);
    cur.push_back(c);
    gen_high(prof, n0, n - c, c, np, cur);
    cur.pop_back();
  }
}

vector<int> qs;
vector<vector<int>> ans;

void solve(const vector<int>& prof, int n0, int n, int c, int p, auto&& cur) {
  for (int i = 0; i < qs.size(); ++i) if (ans[i].empty() && ~high[n0 - n].find(mul(+p, qs[i]))) {
    ans[i] = cur;
    auto& t = high[n0 - n].values[high[n0 - n].find(mul(+p, qs[i]))];
    ans[i].insert(ans[i].end(), t.begin(), t.end());
  }
  int i = 0;
  for (; n + c <= n0 && c <= prof[n + c]; ++c) {
    for (; i < c; ++i) mul(p, n0 - i);
    cur.push_back(c);
    solve(prof, n0, n + c, c, p, cur);
    cur.pop_back();
  }
}

uint64_t eval_high(vector<int> profile) {
  int N = profile.size();
  vector<uint64_t> cnt(N * N);
  cnt[N * N - 1] = 1;
  uint64_t ans = 0;
  for (int n = N; --n; )
  for (int c = N; --c; ) if (cnt[n * N + c]) {
    for (int cc = min(n, c); cc > profile[n]; cc--) {
      cnt[(n - cc) * N + cc] += cnt[n * N + c];
    }
    ans += cnt[n * N + c];
  }
  return ans;
}

uint64_t eval_low(vector<int> profile) {
  int N = profile.size();
  vector<uint64_t> cnt(N * N);
  cnt[0] = 1;
  uint64_t ans = 0;
  for (int n = 0; n < N; ++n)
  for (int c = 0; c <= n; ++c) if (cnt[n * N + c]) {
    for (int cc = max(c, 1); n + cc < N && cc <= profile[n + cc]; ++cc) {
      cnt[(n + cc) * N + cc] += cnt[n * N + c];
    }
    ans += cnt[n * N + c];
  }
  return ans;
}

uint64_t eval(vector<int> profile) {
  return eval_high(profile) + eval_low(profile);
}

vector<int> makeprof(int n, vector<int> from) {
  vector<int> prof(n + 1, 0);
  int j = n;
  for (int i = 0; i < from.size() - 1; ++i) {
    while (j >= from[i]) prof[j--] = i;
  }
  while (j >= 0) prof[j--] = 999;
  return prof;
}

vector<int> getopt(int n) {
  static vector<vector<int>> precalc = [] {
    int n = 121;
    vector<vector<int>> res(n);
    vector<int> tmpl{n, n, n, n, n, n, n, n, n, 0};
    while (n-- > 1) {
      int64_t bst = eval(makeprof(n, tmpl));
      for (int ch = 1; ch--; ) {
        for (int i = 1; i < tmpl.size() - 1; ++i)
        for (auto d: {-1, 1}) if (tmpl[i] + d >= 0 && tmpl[i] + d <= n + 1) {
          tmpl[i] += d;
          int64_t cur = eval(makeprof(n, tmpl));
          if (cur > bst) tmpl[i] -= d;
          else if (cur < bst) ch = 1, bst = cur;
        }
        sort(tmpl.rbegin(), tmpl.rend());
      }
      res[n] = tmpl;
    }
    return res;
  }();
  return makeprof(n, precalc[n]);
}

int main() {
  int tc; cin >> tc;
  qs.resize(tc);
  ans.resize(tc);
  for (int i = 0; i < tc; ++i) {
    cin >> qs[i];
    if (qs[i] == 1) ans[i] = {-1};
    else if (qs[i] == 2) ans[i] = {-1};
    qs[i] = inv(qs[i]);
  }
  for (int n = 1, done = 0; !done++; ++n) {
    inv_.push_back(inv(n));
    high.push_back({});
    for (auto& x: high) x.clear();
    gen_high(getopt(n), n, n, n, 1, vector<int>{});
    solve(getopt(n), n, 0, 1, mul(n + 1, n + 2), vector<int>{});
    cout << n << ' ';
    cout.flush();
    for (int i = 0; i < tc; ++i) done &= !ans[i].empty();
  }
  cout << endl;
  for (int i = 0; i < tc; ++i) {
    if (qs[i] == 1) cout << "1\n";
    else {
      if (ans[i][0] == -1) ans[i] = {};
      cout << accumulate(ans[i].begin(), ans[i].end(), 0) + 2 << "\n1 2\n";
      for (int j = 0, k = 2; j < ans[i].size(); ++j)
      while (ans[i][j]--) cout << j + 1 << ' ' << ++k << '\n';
    }
  }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Wrong Answer
time: 118ms
memory: 3700kb

input:

4
2
360
1
509949433

output:

1 2 3 4 5 6 7 8 
2
1 2
5
1 2
1 3
2 4
2 5
1
10
1 2
1 3
2 4
3 5
4 6
5 7
6 8
7 9
8 10

result:

wrong answer Liczba kolorowań jest błędna!