結果
問題 | No.1667 Forest |
ユーザー |
👑 ![]() |
提出日時 | 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 mathimport sysfrom sys import stdindef modfac(n, MOD):f = 1factorials = [1]for m in range(1, n + 1):f *= mf %= MODfactorials.append(f)inv = pow(f, MOD - 2, MOD)invs = [1] * (n + 1)invs[n] = invfor m in range(n, 1, -1):inv *= minv %= MODinvs[m - 1] = invreturn factorials, invsdef modnCr(n,r,mod,fac,inv):return fac[n] * inv[n-r] * inv[r] % modN,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] = 1for x in range(1,N+1):for y in range(1,N+1):if x < y:now = 0elif x == y:now = 1elif y == 1:now = tree[x]else:now = 0for k in range(1,x):now += tree[k] * modnCr(x-1,k-1,mod,fac,inv) * dp[x-k][y-1]now %= moddp[x][y] = now % modfor M in range(N):print (dp[N][N-M])