結果

問題 No.899 γatheree
コンテスト
ユーザー fumofumofuni
提出日時 2021-04-02 13:07:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,884 bytes
コンパイル時間 2,602 ms
コンパイル使用メモリ 219,748 KB
最終ジャッジ日時 2025-01-20 07:48:51
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0;i<n;i++)
#define repl(i,l,r) for(ll i=(l);i<(r);i++)
#define per(i,n) for(ll i=(n)-1;i>=0;i--)
#define perl(i,r,l) for(ll i=r-1;i>=l;i--)
#define fi first
#define se second
#define pb push_back
#define ins insert
#define pqueue(x) priority_queue<x,vector<x>,greater<x>>
#define all(x) (x).begin(),(x).end()
#define CST(x) cout<<fixed<<setprecision(x)
#define rev(x) reverse(x);
using ll=long long;
using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using pl=pair<ll,ll>;
using vpl=vector<pl>;
using vvpl=vector<vpl>;
const int MOD=1000000007;
const int MOD9=998244353;
const int inf=1e9+10;
const ll INF=1e18;
const ll dy[8]={1,0,-1,0,1,1,-1,-1};
const ll dx[8]={0,1,0,-1,1,-1,1,-1};
template <typename T> inline bool chmax(T &a, T b) {
  return ((a < b) ? (a = b, true) : (false));
}
template <typename T> inline bool chmin(T &a, T b) {
  return ((a > b) ? (a = b, true) : (false));
}

template <typename X, typename M>
struct SegTreeLazy {//遅延セグ木　単位元に注意(updateなら選ばれない数、affineなら(1,0))
    using FX = function<X(X, X)>;
    using FA = function<X(X, M, int)>;
    using FM = function<M(M, M)>;
    int n;
    FX fx;
    FA fa;
    FM fm;
    const X ex;
    const M em;
    vector<X> dat;
    vector<M> lazy;
    SegTreeLazy(int n_, FX fx_, FA fa_, FM fm_, X ex_, M em_)
        : n(), fx(fx_), fa(fa_), fm(fm_), ex(ex_), em(em_), dat(n_ * 4, ex), lazy(n_ * 4, em) {
        int x = 1;
        while (n_ > x) x *= 2;
        n = x;
    }
    void set(int i, X x) { dat[i + n - 1] = x; }
    void build() {
        for (int k = n - 2; k >= 0; k--) dat[k] = fx(dat[2 * k + 1], dat[2 * k + 2]);
    }
    /* lazy eval */
    void eval(int k, int len) {
        if (lazy[k] == em) return;  // 更新するものが無ければ終了
        if (k < n - 1) {            // 葉でなければ子に伝搬
            lazy[k * 2 + 1] = fm(lazy[k * 2 + 1], lazy[k]);
            lazy[k * 2 + 2] = fm(lazy[k * 2 + 2], lazy[k]);
        }
        // 自身を更新
        dat[k] = fa(dat[k],lazy[k],len);//fa(dat[k], fp(lazy[k], len));
        lazy[k] = em;
    }
    void update(int a, int b, M x, int k, int l, int r) {
        eval(k, r - l);
        if (a <= l && r <= b) {  // 完全に内側の時
            lazy[k] = fm(lazy[k], x);
            eval(k, r - l);
        } else if (a < r && l < b) {                     // 一部区間が被る時
            update(a, b, x, k * 2 + 1, l, (l + r) / 2);  // 左の子
            update(a, b, x, k * 2 + 2, (l + r) / 2, r);  // 右の子
            dat[k] = fx(dat[k * 2 + 1], dat[k * 2 + 2]);
        }
    }
    void update(int a, int b, M x) { update(a, b, x, 0, 0, n); }
    X query_sub(int a, int b, int k, int l, int r) {
        eval(k, r - l);
        if (r <= a || b <= l) {  // 完全に外側の時
            return ex;
        } else if (a <= l && r <= b) {  // 完全に内側の時
            return dat[k];
        } else {  // 一部区間が被る時
            X vl = query_sub(a, b, k * 2 + 1, l, (l + r) / 2);
            X vr = query_sub(a, b, k * 2 + 2, (l + r) / 2, r);
            return fx(vl, vr);
        }
    }
    X query(int a, int b) { return query_sub(a, b, 0, 0, n); }
    X operator[](int i){
        return query(i,i+1);
    }
};

vl parent(100010);
vpl child(100010);
vvl g(100010);
vl id(100010,-1);
ll idx=0;
int main(){
  ll n;cin >> n;
  rep(i,n-1){
    ll a,b;cin >> a >> b;
    g[a].pb(b);g[b].pb(a);
  }
  rep(i,n+1)parent[i]=n;
  queue<ll> que;que.push(0);
  while(!que.empty()){
    ll v=que.front();que.pop();
    id[v]=idx;idx++;
    for(auto p:g[v]){
      if(id[p]!=-1)continue;
      parent[p]=v;
      que.push(p);
    }
    child[v]={n,n};
    rep(i,g[v].size()){
      if(g[v][i]==parent[v])continue;
      child[v].se=g[v][i];
    }
    per(i,g[v].size()){
      if(g[v][i]==parent[v])continue;
      child[v].fi=g[v][i];
    }
  }
  id[n]=idx;
  SegTreeLazy<ll,ll> st(n+1,[](ll a,ll b){return a+b;},[](ll a,ll b,int len){return b*len;},[](ll a,ll b){return b;},0,-1);
  rep(i,n){
    ll a;cin >> a;
    st.set(id[i],a);
  }
  st.build();
  ll q;cin >> q;
  while(q--){
    ll x;cin >> x;
    ll sum=st.query(id[child[x].fi],id[child[x].se]+1);
    sum+=st.query(id[child[child[x].fi].fi],id[child[child[x].se].se]+1);
    sum+=st[id[parent[x]]]+st[id[parent[parent[x]]]];
    sum+=st.query(id[child[parent[x]].fi],id[child[parent[x]].se]+1);
    cout << sum <<endl;
    st.update(id[child[x].fi],id[child[x].se]+1,0);
    st.update(id[child[child[x].fi].fi],id[child[child[x].se].se]+1,0);
    st.update(id[child[parent[x]].fi],id[child[parent[x]].se]+1,0);
    st.update(id[x],id[x]+1,sum);st.update(id[parent[x]],id[parent[x]]+1,0);
    st.update(id[parent[parent[x]]],id[parent[parent[x]]]+1,0);
    //rep(i,n)cout << st[id[i]] <<" ";cout << endl;
  }
}
0