結果

問題 No.899 γatheree
ユーザー ks2mks2m
提出日時 2019-10-04 22:46:19
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 1,905 bytes
コンパイル時間 2,432 ms
コンパイル使用メモリ 79,824 KB
実行使用メモリ 50,484 KB
最終ジャッジ日時 2024-04-14 11:22:59
合計ジャッジ時間 7,050 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 52 ms
50,116 KB
testcase_01 AC 58 ms
50,456 KB
testcase_02 AC 58 ms
50,484 KB
testcase_03 AC 57 ms
50,292 KB
testcase_04 AC 58 ms
50,136 KB
testcase_05 AC 57 ms
50,460 KB
testcase_06 TLE -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Queue;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		List<Map<Integer, Integer>> list = new ArrayList<>(n);
		for (int i = 0; i < n; i++) {
			list.add(new HashMap<>());
		}
		for (int i = 0; i < n - 1; i++) {
			String[] sa = br.readLine().split(" ");
			int u = Integer.parseInt(sa[0]);
			int v = Integer.parseInt(sa[1]);
			list.get(u).put(v, 0);
			list.get(v).put(u, 0);
		}
		String[] sa = br.readLine().split(" ");
		long[] a = new long[n];
		for (int i = 0; i < n; i++) {
			a[i] = Integer.parseInt(sa[i]);
		}
		int q = Integer.parseInt(br.readLine());
		int[] x = new int[q];
		for (int i = 0; i < q; i++) {
			x[i] = Integer.parseInt(br.readLine());
		}
		br.close();

		PrintWriter pw = new PrintWriter(System.out);
		for (int i = 0; i < q; i++) {
			Queue<Integer> que = new ArrayDeque<>();
			que.add(x[i]);
			int[] dep = new int[n];
			Arrays.fill(dep, -1);
			dep[x[i]] = 0;
			int[] par = new int[n];
			Arrays.fill(par, -1);
			long sum = a[x[i]];
			while (!que.isEmpty()) {
				Integer cur = que.poll();
				if (dep[cur] < 2) {
					Map<Integer, Integer> cmap = list.get(cur);
					for (int next : cmap.keySet()) {
						if (dep[next] == -1 && cmap.get(next) < 2 - dep[cur]) {
							sum += a[next];
							a[next] = 0;
							que.add(next);
							dep[next] = dep[cur] + 1;
							par[next] = cur;
							cmap.put(next, 2 - dep[cur]);
							list.get(next).put(cur, dep[cur]);
						}
					}
				}
			}
			a[x[i]] = sum;
			pw.println(sum);
		}
		pw.flush();
	}
}
0