結果

問題 No.899 γatheree
コンテスト
ユーザー kotatsugame
提出日時 2019-10-12 03:23:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 760 ms / 2,000 ms
コード長 4,337 bytes
コンパイル時間 1,587 ms
コンパイル使用メモリ 105,584 KB
実行使用メモリ 30,960 KB
最終ジャッジ日時 2024-11-26 09:26:54
合計ジャッジ時間 16,249 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 23
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:127:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  127 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
#include<vector>
#include<functional>
template<typename T>
struct lazysegtree{
	function<T(T,T)>calcfn,lazycalcfn;
	function<T(T,T,unsigned int)>updatefn;
	int n;
	T defvalue,lazydefvalue;
	vector<T>dat,lazy;
	vector<bool>lazyflag;
	lazysegtree(int n_=0,T defvalue_=0,
		function<T(T,T)>calcfn_=[](T a,T b){return a+b;},
		function<T(T,T)>lazycalcfn_=[](T a,T b){return a+b;},
		function<T(T,T,unsigned int)>updatefn_=[](T a,T b,unsigned int width){return a+b*width;},
		T lazydefvalue_=0
	):defvalue(defvalue_),lazydefvalue(lazydefvalue_),
		calcfn(calcfn_),lazycalcfn(lazycalcfn_),updatefn(updatefn_)
	{
		n=1;
		while(n<n_)n<<=1;
		dat.assign(2*n-1,defvalue);
		lazy.assign(2*n-1,lazydefvalue);
		lazyflag.assign(2*n-1,false);
	}
	void copy(const vector<T>&v)
	{
		for(int i=0;i<v.size();i++)dat[i+n-1]=v[i];
		for(int i=n-2;i>=0;i--)dat[i]=calcfn(dat[2*i+1],dat[2*i+2]);
	}
	void eval(int i,int l,int r)
	{
		if(lazyflag[i])
		{
			dat[i]=updatefn(dat[i],lazy[i],r-l);
			if(r-l>1)
			{
				lazy[2*i+1]=lazycalcfn(lazy[2*i+1],lazy[i]);
				lazy[2*i+2]=lazycalcfn(lazy[2*i+2],lazy[i]);
				lazyflag[2*i+1]=lazyflag[2*i+2]=true;
			}
			lazy[i]=lazydefvalue;
			lazyflag[i]=false;
		}
	}
	void update(int a,int b,T x,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return;
		else if(a<=l&&r<=b)
		{
			lazy[k]=lazycalcfn(lazy[k],x);
			lazyflag[k]=true;
			eval(k,l,r);
		}
		else
		{
			update(a,b,x,2*k+1,l,(l+r)/2);
			update(a,b,x,2*k+2,(l+r)/2,r);
			dat[k]=calcfn(dat[2*k+1],dat[2*k+2]);
		}
	}
	T query(int a,int b,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return defvalue;
		else if(a<=l&&r<=b)return dat[k];
		else return calcfn(
			query(a,b,2*k+1,l,(l+r)/2),
			query(a,b,2*k+2,(l+r)/2,r)
		);
	}
};
//Disjoint Sparse Table
#include<vector>
#include<functional>
template<typename T>
struct DST{
	function<T(T,T)>calcfn;
	int n;
	vector<vector<T> >dat;
	DST(const vector<T>&v={},
		function<T(T,T)>calcfn_=[](T a,T b){return a<b?a:b;}
	):calcfn(calcfn_)
	{
		n=v.size();
		dat.push_back(v);
		for(int i=2;i<n;i<<=1)
		{
			dat.push_back(vector<T>(n));
			for(int j=i;j<n;j+=i<<1)
			{
				dat.back()[j-1]=dat[0][j-1];
				for(int k=2;k<=i;k++)
				{
					dat.back()[j-k]=calcfn(dat[0][j-k],dat.back()[j-k+1]);
				}
				dat.back()[j]=dat[0][j];
				for(int k=2;k<=i&&j+k<=n;k++)
				{
					dat.back()[j+k-1]=calcfn(dat[0][j+k-1],dat.back()[j+k-2]);
				}
			}
		}
	}
	T query(int l,int r)const//[l,r)
	{
		if(l<0)l=0;
		if(r>n)r=n;
		r--;
		if(l==r)return dat[0][l];
		int k=31-__builtin_clz(l^r);
		return calcfn(dat[k][l],dat[k][r]);
	}
};
int N;
vector<int>G[1<<17];
int Lc[1<<17],Rc[1<<17];
int p[1<<17];
vector<int>A;
int Ainv[1<<17];
main()
{
	cin>>N;
	for(int i=1;i<N;i++)
	{
		int u,v;cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	for(int i=0;i<N;i++)Lc[i]=Rc[i]=p[i]=-1;
	queue<int>PP;
	PP.push(0);
	A.push_back(0);
	while(!PP.empty())
	{
		int u=PP.front();PP.pop();
		for(int v:G[u])
		{
			if(v==p[u])continue;
			if(Lc[u]<0)Lc[u]=A.size();
			A.push_back(v);
			Rc[u]=A.size();
			p[v]=u;
			PP.push(v);
		}
	}
	for(int i=0;i<N;i++)Ainv[A[i]]=i;
	vector<long>B(N);
	for(int i=0;i<N;i++)cin>>B[Ainv[i]];
	lazysegtree<long>P(N,0,[](long a,long b){return a+b;},
		[](long a,long b){return b;},
		[](long a,long b,unsigned int w){return b*w;});
	vector<int>Lc2(N),Rc2(N);
	for(int i=0;i<N;i++)
	{
		Lc2[i]=Lc[A[i]];
		Rc2[i]=Rc[A[i]];
	}
	DST<int>P2(Lc2,[](int a,int b){return a==-1?b:b==-1?a:a<b?a:b;});
	DST<int>P3(Rc2,[](int a,int b){return a<b?b:a;});
	P.copy(B);
	int Q;cin>>Q;
	for(;Q--;)
	{
		int x;cin>>x;
		long ans=0;
		if(Lc[x]>=0)
		{
			int L=Lc[x],R=Rc[x];
			ans+=P.query(L,R);
			P.update(L,R,0);
			int L2=P2.query(L,R);
			int R2=P3.query(L,R);
			if(L2>=0)
			{
				ans+=P.query(L2,R2);
				P.update(L2,R2,0);
			}
		}
		if(p[x]>=0)
		{
			int u=p[x];
			ans+=P.query(Lc[u],Rc[u]);
			P.update(Lc[u],Rc[u],0);
			if(p[u]>=0)
			{
				int v=p[u];
				ans+=P.query(Ainv[v],Ainv[v]+1);
				P.update(Ainv[v],Ainv[v]+1,0);
			}
			ans+=P.query(Ainv[u],Ainv[u]+1);
			P.update(Ainv[u],Ainv[u]+1,0);
		}
		else
		{
			ans+=P.query(Ainv[x],Ainv[x]+1);
			P.update(Ainv[x],Ainv[x]+1,0);
		}
		cout<<ans<<endl;
		P.update(Ainv[x],Ainv[x]+1,ans);
	}
}
0