結果
| 問題 |
No.899 γatheree
|
| コンテスト | |
| ユーザー |
 smiken_61
|
| 提出日時 | 2019-11-02 19:14:57 |
| 言語 | C++11（廃止可能性あり） (gcc 13.3.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 12,148 bytes |
| コンパイル時間 | 1,798 ms |
| コンパイル使用メモリ | 179,092 KB |
| 実行使用メモリ | 22,528 KB |
| 最終ジャッジ日時 | 2024-09-14 23:14:33 |
| 合計ジャッジ時間 | 10,568 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | WA * 23 |
ソースコード
#include <bits/stdc++.h>
// #include <boost/multiprecision/cpp_int.hpp>
#define int long long
#define inf 1000000007
#define pa pair<int,int>
#define ll long long
#define pal pair<double,double>
#define ppap pair<pa,int>
#define PI 3.14159265358979323846
#define paa pair<int,char>
#define mp make_pair
#define pb push_back
#define EPS (1e-8)
int dx[8]={0,1,0,-1,1,1,-1,-1};
int dy[8]={1,0,-1,0,-1,1,1,-1};
using namespace std;
class pa3{
public:
int x;
int y,z;
pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
bool operator < (const pa3 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
return z<p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa3 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
return z>p.z;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa3 &p) const{
return x==p.x && y==p.y && z==p.z;
}
bool operator != (const pa3 &p) const{
return !( x==p.x && y==p.y && z==p.z);
}
};
class pa4{
public:
int x;
int y,z,w;
pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
bool operator < (const pa4 &p) const{
if(x!=p.x) return x<p.x;
if(y!=p.y) return y<p.y;
if(z!=p.z)return z<p.z;
return w<p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator > (const pa4 &p) const{
if(x!=p.x) return x>p.x;
if(y!=p.y) return y>p.y;
if(z!=p.z)return z>p.z;
return w>p.w;
//return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa4 &p) const{
return x==p.x && y==p.y && z==p.z &&w==p.w;
}
};
class pa2{
public:
int x,y;
pa2(int x=0,int y=0):x(x),y(y) {}
pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
bool operator < (const pa2 &p) const{
return y != p.y ? y<p.y: x<p.x;
}
bool operator > (const pa2 &p) const{
return x != p.x ? x<p.x: y<p.y;
}
bool operator == (const pa2 &p) const{
return abs(x-p.x)==0 && abs(y-p.y)==0;
}
bool operator != (const pa2 &p) const{
return !(abs(x-p.x)==0 && abs(y-p.y)==0);
}
};
string itos( int i ) {
ostringstream s ;
s << i ;
return s.str() ;
}
int gcd(int v,int b){
if(v==0) return b;
if(b==0) return v;
if(v>b) return gcd(b,v);
if(v==b) return b;
if(b%v==0) return v;
return gcd(v,b%v);
}
int mod;
int extgcd(int a, int b, int &x, int &y) {
if (b == 0) {
x = 1;
y = 0;
return a;
}
int d = extgcd(b, a%b, y, x);
y -= a/b * x;
return d;
}
pa operator+(const pa & l,const pa & r) {
return {l.first+r.first,l.second+r.second};
}
pa operator-(const pa & l,const pa & r) {
return {l.first-r.first,l.second-r.second};
}
int pr[1000010];
int inv[1000010];
int beki(int wa,int rr,int warukazu){
if(rr==0) return 1%warukazu;
if(rr==1) return wa%warukazu;
wa%=warukazu;
if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
ll zx=beki(wa,rr/2,warukazu);
return (zx*zx)%warukazu;
}
int comb(int nn,int rr){
if(rr<0 || rr>nn || nn<0) return 0;
int r=pr[nn]*inv[rr];
r%=mod;
r*=inv[nn-rr];
r%=mod;
return r;
}
void gya(int ert){
pr[0]=1;
for(int i=1;i<=ert;i++){
pr[i]=(pr[i-1]*i)%mod;
}
inv[ert]=beki(pr[ert],mod-2,mod);
for(int i=ert-1;i>=0;i--){
inv[i]=inv[i+1]*(i+1)%mod;
}
}
// cin.tie(0);
// ios::sync_with_stdio(false);
//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;
//sort(ve.begin(),ve.end(),greater<int>());
// mt19937(clock_per_sec);
// mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ;
//----------------kokomade tenpure------------
// 範囲を一様にa倍してb足す操作．区間maxminsumに対応
//要はupdate, addにも対応して3種の区間クエリに回答可能
using Node =int;
using Sayo =int;
struct seg_chusho{
// 1
// 2 3
// 4 5 6 7
private:
public:
// (1<<15)=32768
// 1<<16 = 65536
// 1<<17 = 131072
// 1<<18 = 262144
int cor=(1<<18);
/*
struct Node{
int mi;
int ma;
int su;
};
struct Sayo{
int a;
int b;
bool operator==(sayo v){
return a==v.a && b==v.b;
}
};
*/
// 遅延が貯まっていないときに保持する値
Sayo S_id=-1;
Node N_id=0;
Node node_merge(Node x,Node y){
// x と yをマージする 区間和なら和
return x+y;
// ここに実装する
}
Sayo merge_sayo(Sayo x,Sayo y){
// 作用xの上から作用yを更にのっける
if(y==-1) return x;
return y;
// ここに実装する
}
Node node_sayo(Node x,Sayo y,int l,int r,int k){
// 区間[l,r)のNode x に作用yを載せる
// x == N_idの心配は基本的に不要
// y == S_id も不必要
return y*(r-l);
// ここに実装する
}
vector<Node> vec;
vector<Sayo> lazy;
void shoki1(){
//S_id,N_idが未定義ならここでやる
vec.resize(2*cor+3, N_id);
lazy.resize(2*cor+3,S_id);
}
void shoki2(){
for(int i=cor-1;i>0;i--) vec[i]=node_merge(vec[2*i],vec[2*i+1]);
}
void chien(int k,int l,int r){
if(lazy[k]==S_id) return;
if(l+1<r){
vec[k]=node_sayo(vec[k],lazy[k],l,r,k);
lazy[2*k]=merge_sayo(lazy[2*k],lazy[k]);
lazy[2*k+1]=merge_sayo(lazy[2*k+1],lazy[k]);
lazy[k]=S_id;
}
else {
vec[k]=node_sayo(vec[k],lazy[k],l,r,k);
lazy[k]=S_id;
}
}
Node rangeshori(int a,int b,Sayo w, int k=1,int l=0,int r=-3){
if(r<0) r=cor;
if(a<=l && r<=b){
lazy[k]=merge_sayo(lazy[k],w);
chien(k,l,r);
return vec[k];
}
if(r<=a || b<=l){
chien(k,l,r);
return vec[k];
}
chien(k,l,r);
Node w1=rangeshori(a,b,w,k*2,l,(l+r)/2);
Node w2=rangeshori(a,b,w,k*2+1,(l+r)/2,r);
vec[k]=node_merge(w1,w2);
return vec[k];
}
// [a,b)
// k-th node
// k no kukanha [l,r)
Node getrange(int a,int b,int k=1,int l=0,int r=-1){
if(r<0) r=cor;
chien(k,l,r);
if(r<=a || b<=l){// 区間外
return N_id;
}
if(a<=l && r<=b){// 完全含む
return vec[k];
}
Node w1=getrange(a,b,k*2,l,(l+r)/2);
Node w2=getrange(a,b,k*2+1,(l+r)/2,r);
return node_merge(w1,w2);
}
};
seg_chusho ss;
vector<int> G[100020],G2[100020];
int oya[100020];
void dfs(int r,int p){
oya[r]=p;
for(auto v:G[r])if(v!=p){
dfs(v,r);
G2[r].pb(v);
}
}
int iti[100020];
pa ko[100020];
pa kko[100020];
signed main(){
cin.tie(0);
ios::sync_with_stdio(false);
int n;
cin>>n;
for(int i=0;i<n-1;i++){
int y,yy;
cin>>y>>yy;
G[y].pb(yy);
}
dfs(0,0);
queue<int> qu;
qu.push(0);
int cnt=0;
while(qu.size()){
int y=qu.front();
qu.pop();
iti[y]=cnt;
cnt++;
for(auto v:G2[y])qu.push(v);
}
ss.shoki1();
for(int i=0;i<n;i++){
if(G2[i].size()==0)ko[i]=mp(-1,-1);
else{
ko[i].first=iti[G2[i][0]];
ko[i].second=iti[G2[i].back()];
}
}
for(int i=0;i<n;i++){
if(G2[i].size()==0)kko[i]=mp(-1,-1);
else{
int a1=inf,a2=-inf;
for(auto v:G[i]){
if(ko[v].first==-1) continue;
a1=min(a1,ko[v].first);
a2=max(a2,ko[v].second);
}
if(a1==inf)kko[i]=mp(-1,-1);
else{
kko[i]=mp(a1,a2);
}
}
}
for(int i=0;i<n;i++){
int y;
cin>>y;
ss.vec[ss.cor+iti[i]]=y;
}
ss.shoki2();
int q;
cin>>q;
//for(int i=0;i<n;i++)cout<<fis[i]<<" "<<las[i]<<endl;
for(int i=0;i<q;i++){
int p;
cin>>p;
vector<pa> ve;
int x2=oya[oya[p]];
int x1=oya[p];
ve.pb(mp(iti[x2],iti[x2]));
ve.pb(mp(iti[x1],iti[x1]));
if(ko[x1].first!=-1)ve.pb(ko[x1]);
if(ko[p].first!=-1)ve.pb(ko[p]);
if(kko[p].first!=-1)ve.pb(kko[p]);
int ans=0;
for(auto v:ve){
ans+=ss.getrange(v.first,v.second+1);
ss.rangeshori(v.first,v.second+1,0);
}
cout<<ans<<endl;
ss.rangeshori(iti[p],iti[p]+1,ans);
}
return 0;
}