結果

問題 No.899 γatheree
ユーザー niuezniuez
提出日時 2019-09-16 21:38:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 3,278 bytes
コンパイル時間 2,171 ms
コンパイル使用メモリ 187,512 KB
実行使用メモリ 19,280 KB
最終ジャッジ日時 2024-07-07 02:32:32
合計ジャッジ時間 12,569 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
  }
}
0