結果
問題 | No.196 典型DP (1) |
ユーザー | dddgg |
提出日時 | 2024-01-19 15:51:47 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 140 ms / 2,000 ms |
コード長 | 630 bytes |
コンパイル時間 | 203 ms |
コンパイル使用メモリ | 82,092 KB |
実行使用メモリ | 91,604 KB |
最終ジャッジ日時 | 2024-09-28 03:25:27 |
合計ジャッジ時間 | 5,135 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
import sys sys.setrecursionlimit(10 ** 9) MOD = 10 ** 9 + 7 n, k = map(int, input().split()) edges = [[] for _ in range(n)] for _ in range(n - 1): u, v = map(int, input().split()) edges[u].append(v) edges[v].append(u) def dfs(pos, bpos): dp = [1] for npos in edges[pos]: if npos == bpos: continue tmp= dfs(npos, pos) ndp = [0] * (len(dp)+len(tmp)-1) for i, v in enumerate(dp): for j, u in enumerate(tmp): ndp[i + j] += u * v ndp[i + j] %= MOD dp = ndp dp.append(1) return dp dp= dfs(0, -1) print(dp[k])