結果

問題 No.1667 Forest
コンテスト
ユーザー だれ
提出日時 2021-08-26 23:01:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,059 ms / 3,000 ms
コード長 712 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 96,456 KB
最終ジャッジ日時 2024-12-15 08:57:21
合計ジャッジ時間 7,919 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, mod = map(int, input().split())

fact = [1]
for i in range(1, 301):
    fact.append(fact[-1] * i % mod)

inv = [1] * 301
for i in range(1, 301):
    inv[i] = pow(fact[i], mod - 2, mod)

def binom(a, b):
    if a < 0 or b < 0 or a < b:
        return 0
    return fact[a] * inv[b] * inv[a - b] % mod

c = [0, 0]
for i in range(2, 301):
    c.append(pow(i, i - 2, mod))

dp = [[0] * n for _ in range(n + 1)]
dp[0][0] = 1

for i in range(n):
    for j in range(i + 1):
        dp[i + 1][j] += dp[i][j]
        dp[i + 1][j] %= mod
        for k in range(2, n - i + 1):
            dp[i + k][j + k - 1] += dp[i][j] * binom(n - i - 1, k - 1) * c[k]
            dp[i + k][j + k - 1] %= mod

print(*dp[n], sep = "\n")
0