結果

問題 No.899 γatheree
コンテスト
ユーザー polylogK
提出日時 2019-09-17 17:30:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,038 bytes
コンパイル時間 1,445 ms
コンパイル使用メモリ 176,012 KB
実行使用メモリ 18,292 KB
最終ジャッジ日時 2024-07-07 13:54:22
合計ジャッジ時間 4,859 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std::literals::string_literals;
using i64 = long long;
using std::cout;
using std::endl;
using std::cin;

template<typename T>
std::vector<T> make_v(size_t a){return std::vector<T>(a);}

template<typename T,typename... Ts>
auto make_v(size_t a,Ts... ts){
  return std::vector<decltype(make_v<T>(ts...))>(a,make_v<T>(ts...));
}

int main() {
	int n; scanf("%d", &n);
	assert(2 <= n and n <= (int)1e5);
	
	std::vector<std::set<int>> g(n);
	for(int i = 0; i < n - 1; i++) {
		int a, b; scanf("%d%d", &a, &b);

		assert(0 <= a and a < n);
		assert(0 <= b and b < n);
		
		g[a].insert(b);
		g[b].insert(a);
	}

	std::vector<i64> a(n);
	for(int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
		
		assert(0 <= a[i] and a[i] <= (int)1e5);
	}

	int q; scanf("%d", &q);
	assert(1 <= q and q <= (int)1e5);
	
	while(q--) {
		int v; scanf("%d", &v);
		assert(0 <= v and v < n);

		for(auto e: g[v]) {
			a[v] += a[e];
			a[e] = 0;

			g[e].insert(v);
		}
		g[v].clear();

		printf("%lld\n", a[v]);
	}
	return 0;
}
0