結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー kbys
提出日時 2020-03-29 00:35:50
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 632 bytes
コンパイル時間 196 ms
コンパイル使用メモリ 12,032 KB
実行使用メモリ 28,916 KB
最終ジャッジ日時 2025-01-02 12:31:05
合計ジャッジ時間 6,921 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#994
from collections import deque

N,K = map(int,input().split())
edge = [[] for _ in range(N)]
for _ in range(N-1):
    a,b = map(int,input().split())
    edge[a-1].append(b-1)
    edge[b-1].append(a-1)
if N < K:
    print(-1)
else:
    distance = [-1]*N
    distance[0] = 0
    q = deque([0])
    while q:
        now = q.popleft()
        for v in edge[now]:
            if distance[v] != -1:
                continue
            else:
                distance[v] = distance[now]+1
                q.append(v)
    distance.sort()
    dist = distance[1:]
    ans = 0
    for i in range(K-1):
        ans += dist[i]
    print(ans)
0