結果
問題 | No.1978 Permutation Repetition |
ユーザー |
![]() |
提出日時 | 2022-06-03 20:08:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 234 ms / 2,000 ms |
コード長 | 975 bytes |
コンパイル時間 | 212 ms |
コンパイル使用メモリ | 82,184 KB |
実行使用メモリ | 67,456 KB |
最終ジャッジ日時 | 2024-09-21 05:44:42 |
合計ジャッジ時間 | 6,165 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 44 |
ソースコード
from math import gcdMOD = 1000000007N, M = map(int, input().split())A = [int(s) - 1 for s in input().split()]used = [False] * Ncycles = [0] * (N+1)for i in range(N):if used[i]: continuecnt = 0while not used[i]:used[i] = Truei = A[i]cnt += 1cycles[cnt] += 1fact = [0] * (N+1)ifact = [0] * (N+1)fact[0] = 1ifact[0] = 1for i in range(N):fact[i+1] = fact[i] * (i+1) % MODifact[i+1] = pow(fact[i+1], MOD - 2, MOD)def binom(n, k):return fact[n] * ifact[k] % MOD * ifact[n-k] % MODans = 1for i in range(1, N+1):dp = [0] * (cycles[i] + 1)ms = [m for m in range(1, N+1) if m // gcd(m, M) == i]dp[0] = 1for j in range(1, cycles[i] + 1):for m in ms:if j - m//i >= 0:dp[j] += dp[j - m//i] * binom(j-1, m//i - 1) % MOD * fact[m//i - 1] % MOD * pow(i, m//i - 1, MOD) % MODdp[j] %= MODans = ans * dp[cycles[i]] % MODprint(ans)