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

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	mymelochan
提出日時	2022-03-06 15:33:30
言語	PyPy3 (7.3.15)
結果	AC
実行時間	145 ms / 2,000 ms
コード長	852 bytes
コンパイル時間	237 ms
コンパイル使用メモリ	82,472 KB
実行使用メモリ	90,368 KB
最終ジャッジ日時	2024-07-21 00:59:21
合計ジャッジ時間	4,858 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	34 ms 52,352 KB
testcase_01	AC	34 ms 52,224 KB
testcase_02	AC	34 ms 52,460 KB
testcase_03	AC	35 ms 51,840 KB
testcase_04	AC	35 ms 52,224 KB
testcase_05	AC	34 ms 52,096 KB
testcase_06	AC	36 ms 51,968 KB
testcase_07	AC	35 ms 52,224 KB
testcase_08	AC	36 ms 51,712 KB
testcase_09	AC	36 ms 52,224 KB
testcase_10	AC	36 ms 51,968 KB
testcase_11	AC	37 ms 52,480 KB
testcase_12	AC	41 ms 58,368 KB
testcase_13	AC	43 ms 59,904 KB
testcase_14	AC	44 ms 60,416 KB
testcase_15	AC	52 ms 62,592 KB
testcase_16	AC	62 ms 67,840 KB
testcase_17	AC	88 ms 76,540 KB
testcase_18	AC	90 ms 76,612 KB
testcase_19	AC	98 ms 77,352 KB
testcase_20	AC	105 ms 77,312 KB
testcase_21	AC	130 ms 77,696 KB
testcase_22	AC	102 ms 76,900 KB
testcase_23	AC	116 ms 82,688 KB
testcase_24	AC	111 ms 82,048 KB
testcase_25	AC	108 ms 81,024 KB
testcase_26	AC	122 ms 89,728 KB
testcase_27	AC	116 ms 89,728 KB
testcase_28	AC	117 ms 90,368 KB
testcase_29	AC	116 ms 89,856 KB
testcase_30	AC	116 ms 90,240 KB
testcase_31	AC	135 ms 77,568 KB
testcase_32	AC	141 ms 77,048 KB
testcase_33	AC	137 ms 77,676 KB
testcase_34	AC	142 ms 77,184 KB
testcase_35	AC	145 ms 77,064 KB
testcase_36	AC	145 ms 77,104 KB
testcase_37	AC	130 ms 78,028 KB
testcase_38	AC	137 ms 77,340 KB
testcase_39	AC	130 ms 77,048 KB
testcase_40	AC	137 ms 77,644 KB
testcase_41	AC	37 ms 51,840 KB
testcase_42	AC	35 ms 52,096 KB
testcase_43	AC	35 ms 51,968 KB

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

ソースコード

raw source code

import sys
sys.setrecursionlimit(10**4)
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

結果

テストケース

ソースコード