結果
問題 | No.1667 Forest |
ユーザー |
![]() |
提出日時 | 2021-09-03 22:27:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,517 ms / 3,000 ms |
コード長 | 1,200 bytes |
コンパイル時間 | 229 ms |
コンパイル使用メモリ | 82,584 KB |
実行使用メモリ | 81,752 KB |
最終ジャッジ日時 | 2024-12-15 14:41:08 |
合計ジャッジ時間 | 18,187 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
def main(): import sys input = sys.stdin.readline def f(x, mod): if x == 1: return 1 return pow(x, x-2, mod) N, mod = map(int, input().split()) # comb init # mod = 1000000007 nmax = 2 * 10 ** 5 + 10 # change here fac = [0] * nmax finv = [0] * nmax inv = [0] * nmax fac[0] = 1 fac[1] = 1 finv[0] = 1 finv[1] = 1 inv[1] = 1 for i in range(2, nmax): fac[i] = fac[i - 1] * i % mod inv[i] = mod - inv[mod % i] * (mod // i) % mod finv[i] = finv[i - 1] * inv[i] % mod def comb(n, r): if n < r: return 0 else: return (fac[n] * ((finv[r] * finv[n - r]) % mod)) % mod dp = [[0] * N for _ in range(N+1)] dp[N][0] = 1 for i in range(N, 0, -1): for j in range(N): for j_use in range(1, N): if j + j_use < N and i >= j_use + 1: dp[i-(j_use + 1)][j + j_use] = (dp[i-(j_use + 1)][j + j_use] + ((dp[i][j] * f(j_use + 1, mod))%mod * comb(i-1, j_use))%mod)%mod dp[i-1][j] = (dp[i-1][j] + dp[i][j])%mod print(*dp[0], sep="\n") if __name__ == '__main__': main()