結果
問題 | No.1718 Random Squirrel |
ユーザー | tenten |
提出日時 | 2021-10-28 10:00:30 |
言語 | Java21 (openjdk 21) |
結果 |
WA
|
実行時間 | - |
コード長 | 4,683 bytes |
コンパイル時間 | 2,760 ms |
コンパイル使用メモリ | 80,256 KB |
実行使用メモリ | 112,452 KB |
最終ジャッジ日時 | 2024-04-16 13:01:19 |
合計ジャッジ時間 | 25,919 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 54 ms
37,148 KB |
testcase_01 | AC | 55 ms
37,192 KB |
testcase_02 | AC | 54 ms
37,212 KB |
testcase_03 | AC | 55 ms
37,448 KB |
testcase_04 | AC | 56 ms
37,320 KB |
testcase_05 | AC | 55 ms
36,864 KB |
testcase_06 | AC | 53 ms
37,268 KB |
testcase_07 | AC | 53 ms
37,172 KB |
testcase_08 | AC | 53 ms
37,160 KB |
testcase_09 | AC | 56 ms
37,092 KB |
testcase_10 | AC | 53 ms
36,880 KB |
testcase_11 | AC | 751 ms
64,836 KB |
testcase_12 | AC | 373 ms
51,936 KB |
testcase_13 | AC | 686 ms
60,544 KB |
testcase_14 | AC | 769 ms
72,576 KB |
testcase_15 | AC | 1,103 ms
72,236 KB |
testcase_16 | AC | 564 ms
56,024 KB |
testcase_17 | AC | 639 ms
61,592 KB |
testcase_18 | AC | 1,465 ms
87,048 KB |
testcase_19 | AC | 975 ms
67,236 KB |
testcase_20 | AC | 338 ms
51,988 KB |
testcase_21 | AC | 981 ms
71,648 KB |
testcase_22 | AC | 1,552 ms
91,404 KB |
testcase_23 | AC | 1,100 ms
76,600 KB |
testcase_24 | AC | 982 ms
71,652 KB |
testcase_25 | AC | 1,614 ms
92,916 KB |
testcase_26 | AC | 688 ms
74,092 KB |
testcase_27 | AC | 742 ms
74,156 KB |
testcase_28 | AC | 710 ms
73,764 KB |
testcase_29 | WA | - |
testcase_30 | AC | 1,192 ms
112,452 KB |
testcase_31 | AC | 733 ms
95,640 KB |
testcase_32 | WA | - |
ソースコード
import java.io.*; import java.util.*; public class Main { static ArrayList<ArrayList<Integer>> graph = new ArrayList<>(); static ArrayList<ArrayList<Integer>> part = new ArrayList<>(); static int[] counts; static int k; static int base; static int total = 0; public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int n = sc.nextInt(); k = sc.nextInt(); counts = new int[n]; for (int i = 0; i < n; i++) { graph.add(new ArrayList<>()); part.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); } for (int i = 0; i < k; i++) { counts[sc.nextInt() - 1] = 1; } getCount(0, 0); PriorityQueue<Path> queue = new PriorityQueue<>(); int[] costsA = new int[n]; queue.add(new Path(base, 0)); doBTS(queue, costsA); int maxA = 0; int maxAIdx = 0; for (int i = 0; i < n; i++) { if (costsA[i] < Integer.MAX_VALUE && maxA < costsA[i]) { maxA = costsA[i]; maxAIdx = i; } } int[] costsB = new int[n]; queue.add(new Path(maxAIdx, 0)); doBTS(queue, costsB); int maxB = 0; int maxBIdx = 0; for (int i = 0; i < n; i++) { if (costsB[i] < Integer.MAX_VALUE && maxB < costsB[i]) { maxB = costsB[i]; maxBIdx = i; } } int[] costsC = new int[n]; queue.add(new Path(maxBIdx, 0)); doBTS(queue, costsC); int maxC = 0; int maxCIdx = 0; for (int i = 0; i < n; i++) { if (costsC[i] < Integer.MAX_VALUE && maxC < costsC[i]) { maxC = costsC[i]; maxCIdx = i; } } for (int i = 0; i < n; i++) { if (costsA[i] == Integer.MAX_VALUE) { continue; } queue.add(new Path(i, total * 2 - Math.max(costsB[i], costsC[i]))); } int[] ans = new int[n]; Arrays.fill(ans, Integer.MAX_VALUE); while (queue.size() > 0) { Path p = queue.poll(); if (ans[p.idx] <= p.value) { continue; } ans[p.idx] = p.value; for (int x : graph.get(p.idx)) { queue.add(new Path(x, p.value + 1)); } } StringBuilder sb = new StringBuilder(); for (int x : ans) { sb.append(x).append("\n"); } System.out.print(sb); } static int getCount(int idx, int p) { if (counts[idx] == 1) { base = idx; } for (int x : graph.get(idx)) { if (x == p) { continue; } int tmp = getCount(x, idx); if (tmp > 0 && tmp < k) { part.get(idx).add(x); part.get(x).add(idx); total++; } counts[idx] += tmp; } return counts[idx]; } static void doBTS(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 : part.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; } } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public String nextLine() throws Exception { return br.readLine(); } public String next() throws Exception { if (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }