結果
問題 | No.196 典型DP (1) |
ユーザー | rlangevin |
提出日時 | 2023-01-29 16:48:14 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 2,234 bytes |
コンパイル時間 | 146 ms |
コンパイル使用メモリ | 82,112 KB |
実行使用メモリ | 147,896 KB |
最終ジャッジ日時 | 2024-06-29 13:25:29 |
合計ジャッジ時間 | 13,301 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 104 ms
142,020 KB |
testcase_01 | AC | 104 ms
135,124 KB |
testcase_02 | AC | 98 ms
134,968 KB |
testcase_03 | AC | 98 ms
135,112 KB |
testcase_04 | AC | 99 ms
135,044 KB |
testcase_05 | AC | 98 ms
135,008 KB |
testcase_06 | AC | 98 ms
134,944 KB |
testcase_07 | AC | 100 ms
135,056 KB |
testcase_08 | AC | 98 ms
135,156 KB |
testcase_09 | AC | 98 ms
134,996 KB |
testcase_10 | AC | 100 ms
135,136 KB |
testcase_11 | AC | 106 ms
135,168 KB |
testcase_12 | AC | 109 ms
135,384 KB |
testcase_13 | AC | 108 ms
135,060 KB |
testcase_14 | AC | 112 ms
135,148 KB |
testcase_15 | AC | 123 ms
133,900 KB |
testcase_16 | AC | 137 ms
131,012 KB |
testcase_17 | AC | 178 ms
125,120 KB |
testcase_18 | AC | 336 ms
140,708 KB |
testcase_19 | AC | 213 ms
140,820 KB |
testcase_20 | AC | 318 ms
140,696 KB |
testcase_21 | AC | 409 ms
140,796 KB |
testcase_22 | AC | 481 ms
139,940 KB |
testcase_23 | AC | 1,763 ms
141,112 KB |
testcase_24 | TLE | - |
testcase_25 | TLE | - |
testcase_26 | TLE | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
testcase_38 | -- | - |
testcase_39 | -- | - |
testcase_40 | -- | - |
testcase_41 | -- | - |
testcase_42 | -- | - |
testcase_43 | -- | - |
ソースコード
import sys readline = sys.stdin.readline from collections import deque def bfs(G, s, N): Q = deque([]) dist = [-1] * N par = [-1] * N sz = [1] * N dist[s] = 0 for u in G[s]: par[u] = s dist[u] = 1 sz[u] += sz[s] Q.append(u) while Q: u = Q.popleft() for v in G[u]: if dist[v] != -1: continue dist[v] = dist[u] + 1 par[v] = u sz[v] += sz[u] Q.append(v) return par def topsort(G): Q = deque() N = len(G) count = [0] * N for u in range(N): for v in G[u]: count[v] += 1 for u in range(N): if count[u] == 0: Q.append(u) anslst = [] while Q: u = Q.pop() anslst.append(u) for v in G[u]: count[v] -= 1 if count[v] == 0: Q.append(v) return anslst N, K = map(int, input().split()) mod = 10 ** 9 + 7 G = [[] for i in range(N)] a, b = [0] * (N - 1), [0] * (N - 1) for i in range(N - 1): a[i], b[i] = map(int, input().split()) G[a[i]].append(b[i]) G[b[i]].append(a[i]) par = bfs(G, 0, N) G2 = [[] for i in range(N)] for i in range(N - 1): if par[a[i]] == b[i]: G2[a[i]].append(b[i]) else: G2[b[i]].append(a[i]) G = topsort(G2) sz = [1] * N dp0 = [[0] * 2005 for i in range(2005)] dp1 = [[0] * 2005 for i in range(2005)] for i in range(2005): dp1[i][1] = 1 dp0[i][0] = 1 for u in G: for v in G2[u]: pre = sz[v] sz[v] += sz[u] merge0 = [0] * (sz[v] + 1) # merge1 = [0] * (sz[v] + 1) dp0u = dp0[u] dp0v = dp0[v] dp1u = dp1[u] for x in range(sz[u] + 1): d = dp1u[x] + dp0u[x] for y in range(x, sz[v] + 1): # merge1[y] += dp1v[y - x] * dp1u[x] merge0[y] += dp0v[y - x] * (d) # merge1[y] %= mod merge0[y] %= mod for x in range(sz[v] + 1): dp0[v][x] = merge0[x] # dp1[v][x] = merge1[x] dp1[v][sz[v]] = 1 dp1[v][pre] = 0 print( (dp0[0][K] + dp1[0][K]) % mod )