結果
| 問題 |
No.1667 Forest
|
| コンテスト | |
| ユーザー |
👑 SPD_9X2
|
| 提出日時 | 2021-09-03 22:26:43 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,392 ms / 3,000 ms |
| コード長 | 1,629 bytes |
| コンパイル時間 | 298 ms |
| コンパイル使用メモリ | 82,336 KB |
| 実行使用メモリ | 77,208 KB |
| 最終ジャッジ日時 | 2024-12-15 14:36:06 |
| 合計ジャッジ時間 | 10,419 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 15 |
ソースコード
"""
1667:
M辺なので、連結成分の数は固定
n頂点の木の個数は、n**(n-2)
どの頂点をどこに入れるか
を考えなくてはいけない
f(x) = x**(x-2) / fac(x)
として推移させればおk?
dp[x][y] = x頂点を、y個の木にする場合の数
y = 1なら
x**(x-2) である。
そうでないとき、
この中で、最小の数字が含まれる木だけ構築することにする。
k頂点の木の作り方は、
k**(k-2) * nCr(x-1,k-1) である。
これに、dp[x-k][y-1] を掛ければおk
"""
import math
import sys
from sys import stdin
def modfac(n, MOD):
f = 1
factorials = [1]
for m in range(1, n + 1):
f *= m
f %= MOD
factorials.append(f)
inv = pow(f, MOD - 2, MOD)
invs = [1] * (n + 1)
invs[n] = inv
for m in range(n, 1, -1):
inv *= m
inv %= MOD
invs[m - 1] = inv
return factorials, invs
def modnCr(n,r,mod,fac,inv):
return fac[n] * inv[n-r] * inv[r] % mod
N,mod = map(int,stdin.readline().split())
fac,inv = modfac(2*N+100,mod)
tree = [1,1,1]
for i in range(3,400):
tree.append( pow(i,i-2,mod) )
dp = [[0] * (N+1) for i in range(N+1)]
dp[0][0] = 1
for x in range(1,N+1):
for y in range(1,N+1):
if x < y:
now = 0
elif x == y:
now = 1
elif y == 1:
now = tree[x]
else:
now = 0
for k in range(1,x):
now += tree[k] * modnCr(x-1,k-1,mod,fac,inv) * dp[x-k][y-1]
now %= mod
dp[x][y] = now % mod
for M in range(N):
print (dp[N][N-M])
SPD_9X2