結果

問題 No.1030 だんしんぐぱーりない
ユーザー kotatsugame
提出日時 2020-04-22 13:50:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 457 ms / 2,000 ms
コード長 2,055 bytes
コンパイル時間 863 ms
コンパイル使用メモリ 83,540 KB
実行使用メモリ 17,152 KB
最終ジャッジ日時 2024-10-11 07:07:23
合計ジャッジ時間 14,436 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:77:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   77 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#include<vector>
#include<functional>
#include<limits>
template<typename T>
struct segtree{
	using F=function<T(T,T)>;
	const F calcfn,updatefn;
	int n;
	T defvalue;
	vector<T>dat;
	segtree(int n_=0,T defvalue_=numeric_limits<T>::max(),
		const F calcfn_=[](T a,T b){return a<b?a:b;},
		const F updatefn_=[](T a,T b){return b;}
	):defvalue(defvalue_),calcfn(calcfn_),updatefn(updatefn_)
	{
		n=1;
		while(n<n_)n<<=1;
		dat.assign(2*n-1,defvalue);
	}
	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[i*2+1],dat[i*2+2]);
	}
	void update(int i,T a)
	{
		i+=n-1;
		dat[i]=updatefn(dat[i],a);
		while(i>0)
		{
			i=(i-1)/2;
			dat[i]=calcfn(dat[2*i+1],dat[2*i+2]);
		}
	}
	T query(int a,int b)//[a,b)
	{
		int L=(a<0?0:a>n?n:a)+n-1;
		int R=(b<0?0:b>n?n:b)+n-1;
		T lret=defvalue,rret=defvalue;
		for(;L<R;L>>=1,R>>=1)
		{
			if(!(L&1))lret=calcfn(lret,dat[L]);
			if(!(R&1))rret=calcfn(dat[--R],rret);
		}
		return calcfn(lret,rret);
	}
};
int N,K,Q;
int C[1<<17];
int A[1<<17];
vector<int>G[1<<17];
int dep[1<<17];
int pr[17][1<<17];
void dfs(int u,int p,int d,int c)
{
	dep[u]=d;
	pr[0][u]=p;
	C[u]=max(C[u],c);
	for(int v:G[u])dfs(v,u,d+1,C[u]);
}
int lca(int u,int v)
{
	if(dep[u]<dep[v])swap(u,v);
	for(int k=0;k<17;k++)if(dep[u]-dep[v]>>k&1)u=pr[k][u];
	if(u==v)return u;
	for(int k=17;k--;)if(pr[k][u]!=pr[k][v])
	{
		u=pr[k][u];
		v=pr[k][v];
	}
	return pr[0][u];
}
main()
{
	cin>>N>>K>>Q;
	for(int i=0;i<N;i++)cin>>C[i];
	for(int i=0;i<K;i++)cin>>A[i],A[i]--;
	for(int i=1;i<N;i++)
	{
		int u,v;cin>>u>>v;
		u--,v--;
		G[v].push_back(u);
	}
	dfs(0,-1,0,0);
	for(int k=1;k<17;k++)for(int i=0;i<N;i++)
	{
		pr[k][i]=pr[k-1][i]>=0?pr[k-1][pr[k-1][i]]:-1;
	}
	segtree<int>P(K,-1,[](int a,int b){return a==-1?b:b==-1?a:lca(a,b);});
	P.copy(vector<int>(A,A+K));
	for(;Q--;)
	{
		int q,a,b;cin>>q>>a>>b;
		if(q==1)
		{
			a--,b--;
			P.update(a,b);
		}
		else
		{
			cout<<C[P.query(a-1,b)]<<endl;
		}
	}
}
0