#ifdef t9unkubj #pragma GCC optimize("O3") //#include"template.h" #include"template_no_debug.h" #else #undef _GLIBCXX_DEBUG #pragma GCC optimize("O3") #define dbg(...) 199958 using namespace std; #include using uint=unsigned; using ll=long long; using ull=unsigned long long; using ld=long double; using pii=pair; using pll=pair; templateusing vc=vector; templateusing vvc=vc>; templateusing vvvc=vvc>; using vi=vc; using vvi=vc; using vvvi=vc; using vl=vc; using vvl=vc; using vvvl=vc; templateusing smpq=priority_queue,greater>; templateusing bipq=priority_queue; #define rep(i,n) for(ll (i)=0;i<(ll)(n);i++) #define REP(i,j,n) for(ll (i)=(j);i<(ll)(n);i++) #define DREP(i,n,m) for(ll (i)=(n);i>=(m);i--) #define drep(i,n) for(ll i=((n)-1);i>=0;i--) #define all(x) x.begin(),x.end() #define rall(x) x.rbegin(),x.rend() #define mp make_pair #define mt make_tuple #define pb push_back #define eb emplace_back #define fi first #define se second #define is insert #define bg begin() #define ed end() void scan(int&a) { cin >> a; } void scan(ll&a) { cin >> a; } void scan(string&a) { cin >> a; } void scan(char&a) { cin >> a; } void scan(uint&a) { cin >> a; } void scan(ull&a) { cin >> a; } void scan(bool&a) { cin >> a; } void scan(ld&a){ cin>> a;} template void scan(vector&a) { for(auto&x:a) scan(x); } void read() {} template void read(Head&head, Tail&... tail) { scan(head); read(tail...); } #define INT(...) int __VA_ARGS__; read(__VA_ARGS__); #define LL(...) ll __VA_ARGS__; read(__VA_ARGS__); #define ULL(...) ull __VA_ARGS__; read(__VA_ARGS__); #define STR(...) string __VA_ARGS__; read(__VA_ARGS__); #define CHR(...) char __VA_ARGS__; read(__VA_ARGS__); #define DBL(...) double __VA_ARGS__; read(__VA_ARGS__); #define LD(...) ld __VA_ARGS__; read(__VA_ARGS__); #define VC(type, name, ...) vector name(__VA_ARGS__); read(name); #define VVC(type, name, size, ...) vector> name(size, vector(__VA_ARGS__)); read(name); void print(int a) { cout << a; } void print(ll a) { cout << a; } void print(string a) { cout << a; } void print(char a) { cout << a; } void print(uint a) { cout << a; } void print(bool a) { cout << a; } void print(ull a) { cout << a; } void print(ld a){ cout<< a; } template void print(vectora) { for(int i=0;i<(int)a.size();i++){if(i)cout<<" ";print(a[i]);}cout< void PRT(T a) { print(a); cout < void PRT(Head head, Tail ... tail) { print(head); cout << " "; PRT(tail...); return; } template bool chmin(T &x, F y){ if(x>y){ x=y; return true; } return false; } template bool chmax(T &x, F y){ if(x int popcount(T n){ return __builtin_popcount(n); } template T sum(vc&a){ return accumulate(all(a),T(0)); } template T max(vc&a){ return *max_element(all(a)); } template T min(vc&a){ return *min_element(all(a)); } template void unique(vc&a){ a.erase(unique(all(a)),a.end()); } vvi readgraph(int n,int m,int off = -1){ vvi g(n); rep(i, m){ int u,v; cin>>u>>v; u+=off,v+=off; g[u].push_back(v); g[v].push_back(u); } return g; } vvi readtree(int n,int off=-1){ return readgraph(n,n-1,off); } template vc presum(vc &a){ vc ret(a.size()+1); rep(i,a.size())ret[i+1]=ret[i]+a[i]; return ret; } template vc &operator+=(vc &a,F b){ for (auto&v:a)v += b; return a; } template vc &operator-=(vc&a,F b){ for (auto&v:a)v-=b; return a; } template vc &operator*=(vc&a,F b){ for (auto&v:a)v*=b; return a; } #endif double pass_time=0; template struct lazy_segtree{ int n; vectordata; vectorlazy; int log; lazy_segtree(int n1):n(n1){ int n_=1; log=0; while(n_<=n)n_*=2,log++; data=vector(n_*2,e); lazy=vector(n_,id); n=n_; for(int i=n_-1;i>0;i--)update(i); } void update(int k){ data[k]=op(data[k*2],data[k*2+1]); } void apply(int k,F x){// data[k]=mapping(data[k],x); if(k=1;i--){ push(p>>i); } data[p]=x; for(int i=1;i<=log;i++){ update(p>>i); } } void apply(int l,int r,F x){ if(l==r)return; l+=n,r+=n; for(int i=log;i>=1;i--){ if(((l>>i)<>i); if(((r>>i)<>i); } int l2{l},r2{r}; while(l>=1,r>>=1; } l=l2,r=r2; for(int i=1;i<=log;i++){ if(((l>>i)<>i); if(((r>>i)<>i); } } T prod(int l,int r){ l+=n,r+=n; for(int i=log;i>=1;i--){ if(((l>>i)<>i); if(((r>>i)<>i); } T sml=e,smr=e; while(l>=1,r>>=1; } return op(sml,smr); } T get(int p){ p+=n; for(int i=log;i>=1;i--)push(p>>i); return data[p]; } }; template struct bfs_euler_tour{ vector>g; vectorord; vectorpar; vector,dmax+1>>segment; bfs_euler_tour(vector>g):g(g),ord(g.size(),-1),par(g.size(),-1),segment(g.size()){ queueque; que.push(0); int t=0; while(que.size()){ auto p=que.front();que.pop(); ord[p]=t++; for(auto x:g[p]){ if(ord[x]==-1){ par[x]=p; que.push(x); } } } for(int i=0;i>merge_segments(vector>&segments){ sort(segments.begin(),segments.end(),[](auto a,auto b){ return a.second>res; pairnow{1e9,-1};//[a,b) for(auto [a,b]:segments){ if(now.second>=a){//[a,b) [c,d) now.first=min(now.first,a); now.second=max(now.second,b); }else{ if(now!=make_pair(int(1e9),-1))res.emplace_back(now); now=make_pair(a,b); } } res.emplace_back(now); return res; } inline pairvertex(int u,int d){//下にuからの距離がdである頂点集合を返す(半開) if(u==-1)return make_pair(0,1); if(segment[u][d].first==1e9)return make_pair(0,0); return segment[u][d]; } //上に距離がd,上に距離がd-1,..下に距離がdの順に返す(半開) vector>query(int u,int d){ assert(d<=dmax); vector>res; res.emplace_back(make_pair(ord[u],ord[u]+1)); auto u_=u; for(int d_=1;d_<=d;d_++){ if(u_!=-1)u_=par[u_]; res.emplace_back(vertex(u_,d-d_)); res.emplace_back(vertex(u_,0)); } reverse(res.begin(),res.end()); for(int d_=1;d_<=d;d_++){ res.emplace_back(vertex(u,d_)); } return merge_segments(res); } }; ll op(ll a,ll b){return a+b;} ll op2(ll a,int b){if(b)return 0ll;return a;} int op3(int a,int b){return a|b;} void solve(){ INT(n); auto g=readtree(n,0); VC(int,a,n); bfs_euler_tour<2> t(g); lazy_segtreeseg(n); rep(i,n)seg.set(t.ord[i],a[i]); INT(q); while(q--){ INT(x); ll sum=0; auto res=t.query(x,2); for(auto [a,b]:res){ sum+=seg.prod(a,b); seg.apply(a,b,1); } seg.set(t.ord[x],sum); PRT(sum); } } signed main(){ #ifdef t9unkubj freopen("input.txt", "r", stdin); freopen("output.txt", "w", stdout); #endif cin.tie(0)->sync_with_stdio(0); pass_time=clock(); int t=1; while(t--)solve(); pass_time=clock()-pass_time; dbg(pass_time/CLOCKS_PER_SEC); }