結果

問題 No.1978 Permutation Repetition
コンテスト
ユーザー shino16
提出日時 2022-06-03 11:01:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,238 bytes
コンパイル時間 3,455 ms
コンパイル使用メモリ 234,432 KB
最終ジャッジ日時 2025-01-29 17:27:28
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 RE * 1
other AC * 15 RE * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:45:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   45 | //   }
      |   ~~~~ ^               
main.cpp:46:18: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   46 | //   return f;
      |             ~~   ^

ソースコード

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