#ifndef HIDDEN_IN_VS // 折りたたみ用 // 警告の抑制 #define _CRT_SECURE_NO_WARNINGS // ライブラリの読み込み #include using namespace std; // 型名の短縮 using ll = long long; // -2^63 ~ 2^63 = 9 * 10^18(int は -2^31 ~ 2^31 = 2 * 10^9) using pii = pair; using pll = pair; using pil = pair; using pli = pair; using vi = vector; using vvi = vector; using vvvi = vector; using vl = vector; using vvl = vector; using vvvl = vector; using vb = vector; using vvb = vector; using vvvb = vector; using vc = vector; using vvc = vector; using vvvc = vector; using vd = vector; using vvd = vector; using vvvd = vector; template using priority_queue_rev = priority_queue, greater>; using Graph = vvi; // 定数の定義 const double PI = acos(-1); const vi DX = { 1, 0, -1, 0 }; // 4 近傍(下,右,上,左) const vi DY = { 0, 1, 0, -1 }; int INF = 1001001001; ll INFL = 4004004004004004004LL; double EPS = 1e-15; // 入出力高速化 struct fast_io { fast_io() { cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(18); } } fastIOtmp; // 汎用マクロの定義 #define all(a) (a).begin(), (a).end() #define sz(x) ((int)(x).size()) #define lbpos(a, x) (int)distance((a).begin(), std::lower_bound(all(a), x)) #define ubpos(a, x) (int)distance((a).begin(), std::upper_bound(all(a), x)) #define Yes(b) {cout << ((b) ? "Yes\n" : "No\n");} #define YES(b) {cout << ((b) ? "YES\n" : "NO\n");} #define rep(i, n) for(int i = 0, i##_len = int(n); i < i##_len; ++i) // 0 から n-1 まで昇順 #define repi(i, s, t) for(int i = int(s), i##_end = int(t); i <= i##_end; ++i) // s から t まで昇順 #define repir(i, s, t) for(int i = int(s), i##_end = int(t); i >= i##_end; --i) // s から t まで降順 #define repe(v, a) for(const auto& v : (a)) // a の全要素(変更不可能) #define repea(v, a) for(auto& v : (a)) // a の全要素(変更可能) #define repb(set, d) for(int set = 0; set < (1 << int(d)); ++set) // d ビット全探索(昇順) #define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て(昇順) #define smod(n, m) ((((n) % (m)) + (m)) % (m)) // 非負mod #define uniq(a) {sort(all(a)); (a).erase(unique(all(a)), (a).end());} // 重複除去 #define EXIT(a) {cout << (a) << endl; exit(0);} // 強制終了 // 汎用関数の定義 template inline ll pow(T n, int k) { ll v = 1; rep(i, k) v *= n; return v; } template inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す) template inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す) template inline T get(T set, int i) { return (set >> i) & T(1); } // 演算子オーバーロード template inline istream& operator>>(istream& is, pair& p) { is >> p.first >> p.second; return is; } template inline istream& operator>>(istream& is, vector& v) { repea(x, v) is >> x; return is; } template inline vector& operator--(vector& v) { repea(x, v) --x; return v; } template inline vector& operator++(vector& v) { repea(x, v) ++x; return v; } #endif // 折りたたみ用 #if __has_include() #include using namespace atcoder; #ifdef _MSC_VER #include "localACL.hpp" #endif using mint = modint1000000007; //using mint = modint998244353; //using mint = modint; // mint::set_mod(m); namespace atcoder { inline istream& operator>>(istream& is, mint& x) { ll x_; is >> x_; x = x_; return is; } inline ostream& operator<<(ostream& os, const mint& x) { os << x.val(); return os; } } using vm = vector; using vvm = vector; using vvvm = vector; #endif #ifdef _MSC_VER // 手元環境(Visual Studio) #include "local.hpp" #else // 提出用(gcc) inline int popcount(int n) { return __builtin_popcount(n); } inline int popcount(ll n) { return __builtin_popcountll(n); } inline int lsb(int n) { return n != 0 ? __builtin_ctz(n) : -1; } inline int lsb(ll n) { return n != 0 ? __builtin_ctzll(n) : -1; } inline int msb(int n) { return n != 0 ? (31 - __builtin_clz(n)) : -1; } inline int msb(ll n) { return n != 0 ? (63 - __builtin_clzll(n)) : -1; } #define gcd __gcd #define dump(...) #define dumpel(v) #define dump_list(v) #define dump_mat(v) #define input_from_file(f) #define output_to_file(f) #define Assert(b) { if (!(b)) while (1) cout << "OLE"; } #endif //【グラフの入力】O(n + m) /* * (始点, 終点) の組からなる入力を受け取り,n 頂点 m 辺のグラフを構築して返す. * * n : グラフの頂点の数 * m : グラフの辺の数(省略すれば n-1) * undirected : 無向グラフか(省略すれば true) * one_indexed : 入力が 1-indexed か(省略すれば true) */ Graph read_Graph(int n, int m = -1, bool undirected = true, bool one_indexed = true) { // verify : https://codeforces.com/contest/764/problem/C Graph g(n); if (m == -1) m = n - 1; rep(i, m) { int a, b; cin >> a >> b; if (one_indexed) { --a; --b; } g[a].push_back(b); if (undirected) g[b].push_back(a); } return g; } //【木上隣接頂点クエリ(M-可換モノイド)】 /* * Tree_neighbor_vertex_query(Graph g, int rt) : O(n) * rt を根とする根付き木 g と頂点の初期値 v[0..n) = o() で初期化する. * 要素は M-可換モノイド (S, op, o, F, act, comp, id) の元とする. * * Tree_neighbor_vertex_query(Graph g, int rt, vS a) : O(n) * rt を根とする根付き木 g と頂点の初期値 v[0..n) = a[0..n) で初期化する. * * set(int s, S x) : O(log n) * v[s] = x とする. * * S get(int s) : O(log n) * v[s] を返す. * * S child_sum(int s) : O(log n) * 頂点 s の子の値の総和を返す. * * S neighbor_sum(int s) : O(log n) * 頂点 s からの距離が 1 以下である頂点の値の総和を返す. * * S neighbor2_sum(int s) : O(log n) * 頂点 s からの距離が 2 以下である頂点の値の総和を返す. * * S all_sum() : O(1) * 全頂点の値の総和を返す. * * apply(int s, F f) : O(log n) * v[s] に f を作用させる. * * child_apply(int s, F f) : O(log n) * 頂点 s の子の値に f を作用させる. * * neighbor_apply(int s, F f) : O(log n) * 頂点 s からの距離が 1 以下である頂点の値に f を作用させる. * * neighbor2_apply(int s, F f) : O(log n) * 頂点 s からの距離が 2 以下である頂点の値に f を作用させる. * * sort_by_BFS_order(vi& vs) : O(log |vs|) * 頂点集合 vs を BFS 昇順にソートする. */ template class Tree_neighbor_vertex_query { // n : 頂点数 int n; // rt : 根 int rt; // in[s] : 根からの BFS で頂点 s をなぞった時刻(根なら 0) vi in; // in_min[s] : 頂点 s の子 t の in[t] の最小値(葉なら n) // in_max[s] : 頂点 s の子 t の in[t] の最大値(葉なら -1) vi in_min, in_max; // in_min2[s] : 頂点 s の子 t の in_min[t] の最小値(葉なら n) // in_max2[s] : 頂点 s の子 t の in_max[t] の最大値(葉なら -1) vi in_min2, in_max2; // p[s] : 頂点 s の親 vi p; // v[s] : 頂点 s の値 lazy_segtree v; void bfs_euler_tour(const Graph& g, int rt) { // rt から BFS し in[s], p[s] を定める. queue q; q.push(rt); in[rt] = 0; int now = 1; while (!q.empty()) { auto s = q.front(); q.pop(); repe(t, g[s]) { if (in[t] != -1) continue; in[t] = now++; p[t] = s; q.push(t); } } // in_min[s], in_max[s] を定める. rep(s, n) repe(t, g[s]) { if (p[s] == t) continue; chmin(in_min[s], in[t]); chmax(in_max[s], in[t]); } // in_min2[s], in_max2[s] を定める. rep(s, n) repe(t, g[s]) { if (p[s] == t) continue; chmin(in_min2[s], in_min[t]); chmax(in_max2[s], in_max[t]); } } public: // rt を根とする根付き木 g と頂点の初期値 o() で初期化する. Tree_neighbor_vertex_query(const Graph& g, int rt) : n(sz(g)), rt(rt), in(n, -1), in_min(n, n), in_max(n, -1), in_min2(n, n), in_max2(n, -1), p(n, -1), v(n) { bfs_euler_tour(g, rt); } // rt を根とする根付き木 g と頂点の初期値 a[0..n) で初期化する. Tree_neighbor_vertex_query(const Graph& g, int rt, const vector& a) : n(sz(g)), rt(rt), in(n, -1), in_min(n, n), in_max(n, -1), in_min2(n, n), in_max2(n, -1), p(n, -1) { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c bfs_euler_tour(g, rt); vector ini(n); rep(s, n) ini[in[s]] = a[s]; v = lazy_segtree(ini); } Tree_neighbor_vertex_query() : n(0), rt(-1) {} // v[s] = x とする. void set(int s, S x) { Assert(0 <= s && s < n); v.set(in[s], x); } // v[s] を返す. S get(int s) { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c Assert(0 <= s && s < n); return v.get(in[s]); } // 頂点 s の子の値の総和を返す. S child_sum(int s) { Assert(0 <= s && s < n); // s が葉の場合は o() を返す. if (in_max[s] == -1) return o(); return v.prod(in_min[s], in_max[s] + 1); } // 頂点 s からの距離が 1 以下である頂点の値の総和を返す. S neighbor_sum(int s) { Assert(0 <= s && s < n); S res = v.get(in[s]); if (p[s] != -1) res = op(res, v.get(in[p[s]])); res = op(res, child_sum(s)); return res; } // 頂点 s からの距離が 2 以下である頂点の値の総和を返す. S neighbor2_sum(int s) { Assert(0 <= s && s < n); // 根方向に距離 [0..2] 進んだ頂点からの寄与 S res = (p[s] != -1 ? neighbor_sum(p[s]) : v.get(in[s])); // 葉方向に距離 1 進んだ頂点からの寄与 res = op(res, child_sum(s)); // 葉方向に距離 2 進んだ頂点からの寄与 if (in_max2[s] != -1) res = op(res, v.prod(in_min2[s], in_max2[s] + 1)); return res; } // 全頂点の値の総和を返す. S all_sum() { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c return v.all_prod(); } // v[s] に f を作用させる. void apply(int s, F f) { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c Assert(0 <= s && s < n); v.apply(in[s], f); } // 頂点 s の子の値に f を作用させる. void child_apply(int s, F f) { Assert(0 <= s && s < n); // s が葉の場合は何もしない. if (in_max[s] == -1) return; v.apply(in_min[s], in_max[s] + 1, f); } // 頂点 s からの距離が 1 以下である頂点の値に f を作用させる. void neighbor_apply(int s, F f) { Assert(0 <= s && s < n); v.apply(in[s], f); if (p[s] != -1) v.apply(in[p[s]], f); child_apply(s, f); } // 頂点 s からの距離が 2 以下である頂点の値に f を作用させる. void neighbor2_apply(int s, F f) { Assert(0 <= s && s < n); // 根方向に距離 [0..2] 進んだ頂点への作用 if (p[s] != -1) neighbor_apply(p[s], f); else v.apply(in[s], f); // 葉方向に距離 1 進んだ頂点への作用 child_apply(s, f); // 葉方向に距離 2 進んだ頂点への作用 if (in_max2[s] != -1) v.apply(in_min2[s], in_max2[s] + 1, f); } // 全頂点の値に f を作用させる. void all_apply(F f) { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c v.apply(0, n, f); } // 頂点集合 vs を BFS 昇順にソートする. void sort_by_BFS_order(vi& vs) { // verify : https://atcoder.jp/contests/arc148/tasks/arc148_c sort(all(vs), [&](int s, int t) {return in[s] < in[t]; }); } #ifdef _MSC_VER friend ostream& operator<<(ostream& os, Tree_neighbor_vertex_query TNVQ) { os << TNVQ.v; return os; } #endif }; //【変更 作用付き 総和 モノイド】 /* * S ∋ x = {v, c} : c 個の元の和で値 v をとっていることを表す. * F ∋ f : 零次関数 x ← 0 x + f を表す. * x op y : cx + cy 個の元の和で値 vx + vy をとっている状態にする. * f act x : c 個の元の和で値 c f をとっている状態にする. * f comp g : 合成した零次関数 f o g を返す. */ // verify : https://atcoder.jp/contests/abc237/tasks/abc237_g using S109 = pair; // ベクトル (v, c) using F109 = ll; // 行列 (0, f; 0, 1) S109 op109(S109 x, S109 y) { auto [vx, cx] = x; // ベクトル (vx, cx) auto [vy, cy] = y; // ベクトル (vy, cy) // (vx, cx) + (vy, cy) = (vx + vy, cx + cy) return { vx + vy, cx + cy }; } S109 e109() { return { 0, 0 }; } F109 id109() { return INFL + 1; } // 使わない値なら何でも OK S109 act109(F109 f, S109 x) { if (f == id109()) return x; auto [v, c] = x; // ベクトル (v, c) // (0, f; 0, 1).(v, c) = (f c, c) return { f * c, c }; } F109 comp109(F109 f, F109 g) { if (f == id109()) return g; // (0, f; 0, 1).(0, g; 0, 1) = (0, f; 0, 1) return f; } #define Update_Sum_mmonoid S109, op109, e109, F109, act109, comp109, id109 int main() { input_from_file("input.txt"); // output_to_file("output.txt"); int n; cin >> n; auto g = read_Graph(n, n - 1, true, false); vl a(n); cin >> a; vector ini(n); rep(i, n) ini[i] = { a[i], 1 }; Tree_neighbor_vertex_query TNQ(g, 1, ini); int q; cin >> q; rep(hoge, q) { int x; cin >> x; ll val = TNQ.neighbor2_sum(x).first; TNQ.neighbor2_apply(x, 0); TNQ.apply(x, val); cout << val << endl; dump(TNQ); } }