結果
問題 | No.994 ばらばらコイン |
ユーザー |
![]() |
提出日時 | 2022-09-13 18:05:28 |
言語 | Java (openjdk 23) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,887 bytes |
コンパイル時間 | 2,639 ms |
コンパイル使用メモリ | 79,076 KB |
実行使用メモリ | 78,628 KB |
最終ジャッジ日時 | 2024-12-14 01:27:06 |
合計ジャッジ時間 | 9,884 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 7 WA * 16 |
ソースコード
import java.io.*; import java.util.*; public class Main { static ArrayList<ArrayList<Integer>> graph = new ArrayList<>(); static int[] depths; public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int n = sc.nextInt(); int k = sc.nextInt(); if (k > n) { System.out.println(-1); return; } for (int i = 0; i < n; i++) { graph.add(new ArrayList<>()); } for (int i = 0; i < n - 1; i++) { int a = sc.nextInt() - 1; int b = sc.nextInt() - 1; graph.get(a).add(b); graph.get(b).add(a); } depths = new int[n]; setDepth(0, 0, 0); Arrays.sort(depths); long ans = 0; for (int i = 0; i < k; i++) { ans += depths[i]; } System.out.println(ans); } static void setDepth(int idx, int p, int d) { depths[idx] = d; for (int x : graph.get(idx)) { if (x == p) { continue; } setDepth(x, idx, d + 1); } } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); StringBuilder sb = new StringBuilder(); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public double nextDouble() throws Exception { return Double.parseDouble(next()); } public String next() throws Exception { while (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }