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

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	mymelochan
提出日時	2022-03-06 15:33:30
言語	PyPy3 (7.3.15)
結果	AC
実行時間	174 ms / 2,000 ms
コード長	852 bytes
コンパイル時間	800 ms
コンパイル使用メモリ	87,252 KB
実行使用メモリ	91,948 KB
最終ジャッジ日時	2023-09-28 06:33:03
合計ジャッジ時間	7,576 ms
ジャッジサーバーID （参考情報）	judge15 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	73 ms 71,588 KB
testcase_01	AC	75 ms 71,384 KB
testcase_02	AC	74 ms 71,516 KB
testcase_03	AC	74 ms 71,584 KB
testcase_04	AC	74 ms 71,264 KB
testcase_05	AC	72 ms 71,344 KB
testcase_06	AC	73 ms 71,472 KB
testcase_07	AC	72 ms 71,432 KB
testcase_08	AC	73 ms 71,384 KB
testcase_09	AC	72 ms 71,432 KB
testcase_10	AC	74 ms 71,316 KB
testcase_11	AC	75 ms 71,472 KB
testcase_12	AC	76 ms 75,752 KB
testcase_13	AC	77 ms 76,312 KB
testcase_14	AC	79 ms 76,484 KB
testcase_15	AC	83 ms 76,192 KB
testcase_16	AC	98 ms 76,704 KB
testcase_17	AC	120 ms 77,740 KB
testcase_18	AC	124 ms 78,144 KB
testcase_19	AC	131 ms 78,080 KB
testcase_20	AC	135 ms 78,516 KB
testcase_21	AC	168 ms 78,952 KB
testcase_22	AC	136 ms 78,312 KB
testcase_23	AC	156 ms 83,716 KB
testcase_24	AC	154 ms 90,256 KB
testcase_25	AC	146 ms 81,880 KB
testcase_26	AC	153 ms 91,948 KB
testcase_27	AC	152 ms 90,132 KB
testcase_28	AC	154 ms 90,148 KB
testcase_29	AC	152 ms 90,588 KB
testcase_30	AC	153 ms 90,228 KB
testcase_31	AC	173 ms 78,256 KB
testcase_32	AC	173 ms 78,260 KB
testcase_33	AC	170 ms 78,352 KB
testcase_34	AC	173 ms 78,204 KB
testcase_35	AC	172 ms 78,424 KB
testcase_36	AC	174 ms 78,468 KB
testcase_37	AC	167 ms 79,016 KB
testcase_38	AC	170 ms 78,328 KB
testcase_39	AC	166 ms 78,396 KB
testcase_40	AC	172 ms 78,352 KB
testcase_41	AC	72 ms 71,428 KB
testcase_42	AC	71 ms 71,432 KB
testcase_43	AC	72 ms 71,308 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

結果

テストケース

ソースコード