結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー neterukun
提出日時 2020-02-21 21:29:48
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 624 bytes
コンパイル時間 379 ms
コンパイル使用メモリ 82,836 KB
実行使用メモリ 97,468 KB
最終ジャッジ日時 2024-10-08 21:10:38
合計ジャッジ時間 4,036 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque


n, k = list(map(int, input().split()))
info = [list(map(int, input().split())) for i in range(n - 1)]

tree = [[] for i in range(n)]
for i in range(n - 1):
    a, b = info[i]
    a -= 1
    b -= 1
    tree[a].append(b)
    tree[b].append(a)
    
if n < k:
    print(-1)

start = 0
visited = [-1] * n
q = deque([0])
visited[0] = 0
while q:
    pos = q.popleft()
    for next_pos in tree[pos]:
        if visited[next_pos] >= 0:
            continue
        else:
            visited[next_pos] = visited[pos] + 1
            q.append(next_pos)
visited = sorted(visited)
print(sum(visited[0:k]))
0