結果
問題 | No.806 木を道に |
ユーザー |
![]() |
提出日時 | 2020-06-10 13:49:35 |
言語 | Java (openjdk 23) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,579 bytes |
コンパイル時間 | 3,056 ms |
コンパイル使用メモリ | 87,384 KB |
実行使用メモリ | 85,712 KB |
最終ジャッジ日時 | 2024-06-23 03:47:59 |
合計ジャッジ時間 | 23,400 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 9 WA * 18 |
ソースコード
import java.util.*; public class Main { public static void main (String[] args) { Scanner sc = new Scanner(System.in); int n = sc.nextInt(); ArrayList<ArrayList<Integer>> graph = new ArrayList<>(); 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); } PriorityQueue<Path> queue = new PriorityQueue<>(); int[] costs = new int[n]; queue.add(new Path(0, 0)); search(graph, queue, costs); int max = 0; int maxIdx = 0; for (int i = 0; i < n; i++) { if (max < costs[i]) { max = costs[i]; maxIdx = i; } } queue.add(new Path(maxIdx, 0)); search(graph, queue, costs); Arrays.sort(costs); System.out.println(n - costs[n - 1] - 1); } static void search(ArrayList<ArrayList<Integer>> graph, PriorityQueue<Path> queue, int[] costs) { Arrays.fill(costs, Integer.MAX_VALUE); while (queue.size() > 0) { Path p = queue.poll(); if (costs[p.idx] <= p.value) { continue; } costs[p.idx] = p.value; for (int x : graph.get(p.idx)) { queue.add(new Path(x, p.value + 1)); } } } static class Path implements Comparable<Path> { int idx; int value; public Path(int idx, int value) { this.idx = idx; this.value = value; } public int compareTo(Path another) { return value - another.value; } } }