結果

問題 No.899 γatheree
ユーザー fumofumofunifumofumofuni
提出日時 2021-04-02 15:55:52
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 916 ms / 2,000 ms
コード長 5,077 bytes
コンパイル時間 2,895 ms
コンパイル使用メモリ 223,452 KB
実行使用メモリ 20,500 KB
最終ジャッジ日時 2023-08-25 00:10:56
合計ジャッジ時間 21,925 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 6 ms
8,504 KB
testcase_01 AC 6 ms
8,452 KB
testcase_02 AC 6 ms
8,456 KB
testcase_03 AC 6 ms
8,564 KB
testcase_04 AC 6 ms
8,500 KB
testcase_05 AC 6 ms
8,360 KB
testcase_06 AC 843 ms
20,220 KB
testcase_07 AC 901 ms
20,172 KB
testcase_08 AC 908 ms
20,100 KB
testcase_09 AC 916 ms
20,152 KB
testcase_10 AC 905 ms
20,176 KB
testcase_11 AC 908 ms
20,184 KB
testcase_12 AC 861 ms
20,084 KB
testcase_13 AC 855 ms
20,220 KB
testcase_14 AC 852 ms
20,196 KB
testcase_15 AC 882 ms
20,180 KB
testcase_16 AC 899 ms
20,168 KB
testcase_17 AC 850 ms
20,024 KB
testcase_18 AC 875 ms
20,128 KB
testcase_19 AC 910 ms
20,088 KB
testcase_20 AC 915 ms
20,232 KB
testcase_21 AC 874 ms
20,252 KB
testcase_22 AC 882 ms
20,196 KB
testcase_23 AC 894 ms
20,500 KB
権限があれば一括ダウンロードができます

ソースコード

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];
    }
  }
  child[n]={n,n};
  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;
  vector<pl> ccc(n+1,{INF,-INF});
  rep(i,n){
    ll to=parent[i];to=parent[to];
    chmin(ccc[to].fi,id[i]);
    chmax(ccc[to].se,id[i]);
  }
  rep(i,n+1)if(ccc[i].fi==INF)ccc[i]={n,n};
  while(q--){
    ll x;cin >> x;
    ll sum=st.query(id[child[x].fi],id[child[x].se]+1);
    sum+=st.query(ccc[x].fi,ccc[x].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);
    if(parent[x]==n){
      sum+=st[id[x]];
    }
    cout << sum <<endl;
    st.update(id[child[x].fi],id[child[x].se]+1,0);
    st.update(ccc[x].fi,ccc[x].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+1)cout << st[id[i]] <<" ";cout << endl;
  }
}
0