#743930 (PyPy3) No.196 典型DP (1)

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	mymelochan
提出日時	2022-03-06 15:33:02
言語	PyPy3 (7.3.15)
結果	RE
実行時間	-
コード長	812 bytes
コンパイル時間	149 ms
コンパイル使用メモリ	81,792 KB
実行使用メモリ	82,304 KB
最終ジャッジ日時	2024-07-21 00:59:15
合計ジャッジ時間	5,346 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	34 ms 52,224 KB
testcase_01	AC	35 ms 52,096 KB
testcase_02	AC	36 ms 52,224 KB
testcase_03	AC	36 ms 52,096 KB
testcase_04	AC	34 ms 51,840 KB
testcase_05	AC	37 ms 52,224 KB
testcase_06	AC	36 ms 52,224 KB
testcase_07	AC	34 ms 51,968 KB
testcase_08	AC	33 ms 52,480 KB
testcase_09	AC	34 ms 51,968 KB
testcase_10	AC	34 ms 52,352 KB
testcase_11	AC	35 ms 52,352 KB
testcase_12	AC	42 ms 58,496 KB
testcase_13	AC	43 ms 59,648 KB
testcase_14	AC	44 ms 60,800 KB
testcase_15	AC	50 ms 62,336 KB
testcase_16	AC	61 ms 67,328 KB
testcase_17	AC	86 ms 76,456 KB
testcase_18	AC	89 ms 76,764 KB
testcase_19	AC	96 ms 77,224 KB
testcase_20	AC	103 ms 77,380 KB
testcase_21	AC	125 ms 77,312 KB
testcase_22	AC	100 ms 76,800 KB
testcase_23	AC	119 ms 82,304 KB
testcase_24	AC	111 ms 81,920 KB
testcase_25	AC	108 ms 81,024 KB
testcase_26	RE	-
testcase_27	RE	-
testcase_28	RE	-
testcase_29	RE	-
testcase_30	RE	-
testcase_31	AC	144 ms 77,184 KB
testcase_32	AC	143 ms 77,428 KB
testcase_33	AC	135 ms 77,184 KB
testcase_34	AC	140 ms 77,184 KB
testcase_35	AC	136 ms 77,440 KB
testcase_36	AC	136 ms 77,108 KB
testcase_37	AC	131 ms 77,744 KB
testcase_38	AC	131 ms 77,184 KB
testcase_39	AC	131 ms 77,440 KB
testcase_40	AC	135 ms 77,284 KB
testcase_41	AC	37 ms 51,968 KB
testcase_42	AC	35 ms 52,096 KB
testcase_43	AC	34 ms 52,224 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

N,K = map(int,input().split())
G = [[] for _ in range(N)]
for _ in range(N-1):
    u,v = map(int,input().split())
    G[u].append(v)
    G[v].append(u)

visited = [0]*N
size = [0]*N
dp = [[] for _ in range(N)] #dp[v][k]:vの部分木でk個黒が塗られている数

MOD = 10**9+7
def dfs(cur):
    visited[cur] = 1
    size[cur] = 1
    dp_cur = [1]
    for nxt in G[cur]:
        if visited[nxt]:
            continue
        dfs(nxt)
        dp_nxt = dp[nxt]
        ndp = [0]*(size[cur]+size[nxt]-1)
        for i in range(size[cur]):
            for j in range(size[nxt]):
                ndp[i+j] += dp_cur[i]*dp_nxt[j]
                ndp[i+j] %= MOD
        dp_cur = ndp
        size[cur] += (size[nxt]-1)
    dp_cur.append(1)
    size[cur] += 1
    dp[cur] = dp_cur

dfs(0)
print(dp[0][K])
#print(dp)

yukicoder

結果

テストケース

ソースコード