結果
問題 | No.1667 Forest |
ユーザー |
![]() |
提出日時 | 2021-09-03 22:45:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 882 ms / 3,000 ms |
コード長 | 780 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 82,524 KB |
実行使用メモリ | 153,976 KB |
最終ジャッジ日時 | 2024-12-15 15:51:32 |
合計ジャッジ時間 | 6,866 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
n, MOD = map(int, input().split()) nCr = [[0] * (n + 1) for _ in range(n + 1)] nCr[0][0] = 1 for i in range(n + 1): for j in range(n + 1): nCr[i][j] %= MOD if i < n: nCr[i + 1][j] += nCr[i][j] if j < n: nCr[i + 1][j + 1] += nCr[i][j] dp = [[0] * (n + 1) for _ in range(n + 1)] dp[0][0] = 1 powk = [0] * (n + 1) powk[1] = 1 for i in range(2, n + 1): powk[i] = pow(i, i - 2, MOD) for i in range(n): for j in range(n): dp[i][j] %= MOD for k in range(1, n + 1 - i): if k == 1: dp[i + k][j + 1] += dp[i][j] * nCr[n - i - 1][k - 1] else: dp[i + k][j + 1] += dp[i][j] * nCr[n - i - 1][k - 1] * powk[k] for x in dp[n][::-1][:-1]: print(x % MOD)