結果

問題 No.1978 Permutation Repetition
ユーザー shino16shino16
提出日時 2022-06-03 11:01:58
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 2,238 bytes
コンパイル時間 3,268 ms
コンパイル使用メモリ 243,816 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-21 05:44:20
合計ジャッジ時間 7,907 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 AC 2 ms
6,940 KB
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 AC 2 ms
6,944 KB
testcase_06 RE -
testcase_07 RE -
testcase_08 AC 2 ms
6,944 KB
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 AC 2 ms
6,944 KB
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 AC 1 ms
6,940 KB
testcase_24 RE -
testcase_25 AC 2 ms
6,944 KB
testcase_26 AC 1 ms
6,940 KB
testcase_27 RE -
testcase_28 RE -
testcase_29 AC 2 ms
6,940 KB
testcase_30 AC 2 ms
6,944 KB
testcase_31 AC 2 ms
6,940 KB
testcase_32 AC 2 ms
6,944 KB
testcase_33 AC 2 ms
6,940 KB
testcase_34 AC 2 ms
6,940 KB
testcase_35 RE -
testcase_36 AC 2 ms
6,944 KB
testcase_37 RE -
testcase_38 AC 2 ms
6,944 KB
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 AC 2 ms
6,944 KB
testcase_45 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 2 "lib/prelude.hpp"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep2(i, m, n) for (auto i = (m); i < (n); i++)
#define rep(i, n) rep2(i, 0, n)
#define repr2(i, m, n) for (auto i = (n); i-- > (m);)
#define repr(i, n) repr2(i, 0, n)
#define all(x) begin(x), end(x)
#line 2 "main.cpp"
#include <atcoder/convolution>
using mint = atcoder::modint1000000007;

vector<mint> conv(vector<mint> a, vector<mint> b, int deg = -1) {
  if (~deg)
    a.resize(min<int>(a.size(), deg)),
      b.resize(min<int>(b.size(), deg));
  auto f = convolution(a, b);
  if (~deg) f.resize(deg);
  return f;
}

// vector<mint> exp(vector<mint> h, int deg) {
//   assert(h[0] == 0);
//   int n = 1 << atcoder::internal::ceil_pow2(deg);
//   h.resize(n);
//   vector<mint> f = {1}, g = {1};
//   vector<mint> q(h.size() - 1);
//   rep(i, q.size()) q[i] = h[i + 1] * (i + 1);
//   for (int m = 1; m < n; m *= 2) {
//     auto fgg = conv(conv(f, g, m), g, m);
//     g.resize(m);
//     rep(i, m) g[i] = g[i] * 2 - fgg[i];
//     auto fq = conv(f, {q.begin(), q.begin() + m - 1});
//     rep(i, fq.size()) fq[i] =
//       -fq[i] + (i < f.size() - 1 ? f[i + 1] * (i + 1) : 0);
//     auto w = conv(g, fq, m * 2 - 1);
//     rep(i, m - 1) w[i] += q[i];
//     w.push_back(0);
//     repr(i, m * 2 - 1) w[i + 1] = h[i + 1] - w[i] / (i + 1);
//     w[0] = h[0];
//     auto fw = conv(f, w, m * 2);
//     f.resize(m * 2);
//     rep(i, m * 2) f[i] += fw[i];
//   }
//   return f;
// }

int N, M, A[1000];
bool used[1000];
int cycles[1001];

int main() {
  scanf("%d%d", &N, &M);
  rep(i, N) scanf("%d", A+i), A[i]--;

  rep(i, N) if (!used[i]) {
    int cnt = 0;
    for (int v = i; !used[v]; v = A[v])
      used[v] = true, cnt++;
    cycles[cnt]++;
  }

  mint ans = 1;
  rep(m, N+1) if (cycles[m] != 0) {
    int l = cycles[m], g = gcd(M, m);

    vector<mint> poly(l+1);
    rep2(k, 1, l+1) if (M % k == 0 && k % g == 0) {
      poly[k] = mint(m).pow(k-1) / k;
    }
    vector<mint> po = {1};
    vector<mint> exp(l+1, 0);
    mint ifact = 1;
    rep(i, l) {
      po = conv(move(po), poly, l+1);
      ifact /= i+1;
      rep(j, l+1) exp[j] += po[j] * ifact;
    }
    ans *= exp[l] / ifact;
  }

  printf("%d\n", ans.val());
}
0