結果
問題 | No.899 γatheree |
ユーザー | niuez |
提出日時 | 2019-09-16 21:38:33 |
言語 | C++14 (gcc 13.2.0 + boost 1.83.0) |
結果 |
WA
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 3,278 bytes |
コンパイル時間 | 2,600 ms |
コンパイル使用メモリ | 184,660 KB |
実行使用メモリ | 19,184 KB |
最終ジャッジ日時 | 2023-09-21 08:07:02 |
合計ジャッジ時間 | 15,296 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | WA | - |
testcase_01 | WA | - |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | WA | - |
testcase_23 | WA | - |
ソースコード
#include <bits/stdc++.h> using namespace std; using i64 = long long; #define rep(i,s,e) for(i64 (i) = (s);(i) < (e);(i)++) #define all(x) x.begin(),x.end() #define let auto const struct lazy_segment_tree { using T = i64; using L = i64; T t_ide() { return 0; } L l_ide() { return 0; } T ope(const T& a, const T& b) { return a + b; } L lazy_ope(const L& a, const L& b) { return b; } T effect(const T& t, const L& l, const i64 len) { return l * len; } i64 n; vector<T> node; vector<L> lazy; vector<bool> flag; /* create lazy_segment_tree for sequence [t_ide; sz]A */ lazy_segment_tree(vector<T> init) { n = 1; while(n < init.size()) n *= 2; node.assign(n * 2, t_ide()); lazy.assign(n * 2, l_ide()); flag.assign(n * 2, false); for(int i = 0;i < init.size();i++) node[i + n - 1] = init[i]; for(int i = n - 2;i >= 0;i--) node[i] = ope(node[i * 2 + 1], node[i * 2 + 2]); } void eval(i64 l, i64 r, i64 k) { if(flag[k]) { node[k] = effect(node[k], lazy[k], r - l); if(r - l > 1) { lazy[k * 2 + 1] = lazy_ope(lazy[k * 2 + 1], lazy[k]); flag[k * 2 + 1] = true; lazy[k * 2 + 2] = lazy_ope(lazy[k * 2 + 2], lazy[k]); flag[k * 2 + 2] = true; } flag[k] = false; lazy[k] = l_ide(); } } /* for i in [a, b) effect(a[i], lx) */ void update(i64 a, i64 b, L lx, i64 l = 0, i64 r = -1, i64 k = 0) { if(r < 0) r = n; eval(l, r, k); if(b <= l || r <= a) return; else if(a <= l && r <= b) { lazy[k] = lazy_ope(lazy[k], lx); flag[k] = true; eval(l, r, k); } else { update(a, b, lx, l, (l + r) / 2, k * 2 + 1); update(a, b, lx, (l + r) / 2, r, k * 2 + 2); node[k] =ope(node[k * 2 + 1], node[k * 2 + 2]); } } /* the sum of a[i] for i in [a, b) */ T get(i64 a, i64 b, i64 l = 0, i64 r = -1, i64 k = 0) { if(r < 0) r = n; eval(l, r, k); if(b <= l || r <= a) return t_ide(); else if(a <= l && r <= b) return node[k]; else return ope(get(a, b, l, (l + r) / 2, k * 2 + 1), get(a, b, (l + r) / 2, r, k * 2 + 2)); } }; vector<i64> vec; vector<i64> idx; vector<i64> L; vector<i64> R; vector<i64> p; vector<vector<i64>> G; void dfs(i64 v, i64 f) { p[v] = f; L[v] = vec.size(); for(auto t: G[v]) { if(t == f) continue; idx[t] = vec.size(); vec.push_back(t); } R[v] = vec.size(); for(auto t: G[v]) { if(t == f) continue; dfs(t, v); } } int main() { i64 N; cin >> N; idx.resize(N); L.resize(N); R.resize(N); p.resize(N, -1); G.resize(N); for(int i = 0;i < N - 1; i++) { i64 a, b; cin >> a >> b; G[a].push_back(b); G[b].push_back(a); } vec.push_back(0); idx[0] = 0; dfs(0, -1); vector<i64> A(N); for(int i = 0; i < N; i++) { cin >> A[idx[i]]; } lazy_segment_tree seg(A); i64 Q; cin >> Q; for(int q = 0; q < Q; q++) { i64 x; cin >> x; i64 nei = seg.get(L[x], R[x]); if(p[x] != -1) { nei += seg.get(idx[p[x]], idx[p[x]] + 1); } i64 now = seg.get(idx[x], idx[x] + 1); now += nei; seg.update(idx[x], idx[x] + 1, now); seg.update(L[x], R[x], 0); if(p[x] != -1) seg.update(idx[p[x]], idx[p[x]] + 1, 0); cout << now << endl; } }