結果

問題 No.900 aδδitivee
ユーザー kotatsugame
提出日時 2019-10-12 03:53:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 345 ms / 2,000 ms
コード長 2,744 bytes
コンパイル時間 1,074 ms
コンパイル使用メモリ 93,680 KB
実行使用メモリ 27,452 KB
最終ジャッジ日時 2024-11-26 10:24:11
合計ジャッジ時間 11,080 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 27
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:98:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   98 | 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,int,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,int,int)>updatefn_=[](T a,T b,int l,int r){return a+b*(r-l);},
		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],l,r);
			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)
		);
	}
};
int N;
vector<pair<int,int> >G[1<<17];
int L[1<<17],R[1<<17];
vector<long>A;
vector<int>sign,sum;
void dfs(int u,int p)
{
	L[u]=A.size();
	for(pair<int,int>q:G[u])
	{
		if(q.first==p)continue;
		A.push_back(q.second);
		sign.push_back(1);
		dfs(q.first,u);
		A.push_back(-q.second);
		sign.push_back(-1);
	}
	R[u]=A.size();
}
main()
{
	cin>>N;
	for(int i=1;i<N;i++)
	{
		int u,v,w;cin>>u>>v>>w;
		G[u].push_back(make_pair(v,w));
		G[v].push_back(make_pair(u,w));
	}
	dfs(0,-1);
	sum.push_back(0);
	for(int s:sign)sum.push_back(sum.back()+s);
	lazysegtree<long>P(A.size(),0,[](long a,long b){return a+b;},
	[](long a,long b){return a+b;},
	[](long a,long b,int l,int r){return a+b*(sum[r]-sum[l]);});
	P.copy(A);
	int Q;cin>>Q;
	for(;Q--;)
	{
		int q;cin>>q;
		if(q==1)
		{
			int a;long x;
			cin>>a>>x;
			P.update(L[a],R[a],x);
		}
		else
		{
			int b;cin>>b;
			cout<<P.query(0,L[b])<<endl;
		}
	}
}
0