結果
問題 |
No.994 ばらばらコイン
|
ユーザー |
![]() |
提出日時 | 2024-07-03 08:55:54 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 653 bytes |
コンパイル時間 | 335 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 92,032 KB |
最終ジャッジ日時 | 2024-07-03 08:55:58 |
合計ジャッジ時間 | 3,919 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 7 WA * 16 |
ソースコード
import sys input = sys.stdin.readline from collections import deque def bfs(G, s): Q = deque([s]) N = len(G) dist = [-1] * N par = [-1] * N dist[s] = 0 while Q: u = Q.popleft() for v in G[u]: if dist[v] != -1: continue dist[v] = dist[u] + 1 par[v] = u Q.append(v) return dist N, K = map(int, input().split()) G = [[] for i in range(N)] for i in range(N - 1): a, b = map(int, input().split()) a, b = a - 1, b - 1 G[a].append(b) G[b].append(a) if K >= N + 1: print(-1) exit() print(sum(sorted(bfs(G, 0))[:K]))