結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー mokrai
提出日時 2020-02-22 00:57:19
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 651 bytes
コンパイル時間 189 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 821,248 KB
最終ジャッジ日時 2024-10-09 06:12:11
合計ジャッジ時間 3,482 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 WA * 2 MLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, K = list(map(int, input().split()))
ab = []
for i in range(N-1):
    a, b = list(map(int, input().split()))
    ab.append([a-1, b-1])

if N < K:
    print(-1)
    exit()

G = []
for i in range(N):
    G.append([False] * N)

for a, b in ab:
    G[a][b] = True
    G[b][a] = True

# for i in range(N):
#     print(G[i])

VISITED = [None] * N

DEPTH = [None] * N

def dfs(p, d):
    VISITED[p] = True
    DEPTH[p] = d

    # if all(VISITED):
    #     return

    for i, b in enumerate(G[p]):
        if b and not VISITED[i]:
            dfs(i, d + 1)

dfs(0, 0)

# print(':', VISITED)
# print(':', DEPTH)

DEPTH.sort()
ans = sum(DEPTH[:K])
print(ans)
0