結果
問題 | No.994 ばらばらコイン |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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]))