結果

問題 No.1030 だんしんぐぱーりない
ユーザー chocopuuchocopuu
提出日時 2020-04-19 16:20:15
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 373 ms / 2,000 ms
コード長 4,102 bytes
コンパイル時間 2,617 ms
コンパイル使用メモリ 214,820 KB
実行使用メモリ 18,228 KB
最終ジャッジ日時 2024-10-05 16:12:12
合計ジャッジ時間 14,226 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 3 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 317 ms
15,944 KB
testcase_06 AC 223 ms
12,628 KB
testcase_07 AC 139 ms
7,296 KB
testcase_08 AC 148 ms
8,960 KB
testcase_09 AC 226 ms
16,112 KB
testcase_10 AC 92 ms
5,248 KB
testcase_11 AC 233 ms
10,112 KB
testcase_12 AC 233 ms
14,460 KB
testcase_13 AC 201 ms
13,224 KB
testcase_14 AC 270 ms
13,220 KB
testcase_15 AC 113 ms
5,248 KB
testcase_16 AC 228 ms
11,684 KB
testcase_17 AC 230 ms
16,296 KB
testcase_18 AC 290 ms
13,976 KB
testcase_19 AC 150 ms
6,912 KB
testcase_20 AC 182 ms
9,984 KB
testcase_21 AC 181 ms
12,120 KB
testcase_22 AC 181 ms
10,496 KB
testcase_23 AC 222 ms
9,720 KB
testcase_24 AC 161 ms
6,656 KB
testcase_25 AC 201 ms
9,728 KB
testcase_26 AC 123 ms
5,248 KB
testcase_27 AC 138 ms
5,248 KB
testcase_28 AC 244 ms
11,876 KB
testcase_29 AC 179 ms
14,100 KB
testcase_30 AC 167 ms
10,624 KB
testcase_31 AC 179 ms
9,728 KB
testcase_32 AC 225 ms
12,872 KB
testcase_33 AC 236 ms
14,780 KB
testcase_34 AC 93 ms
5,760 KB
testcase_35 AC 373 ms
18,104 KB
testcase_36 AC 355 ms
18,100 KB
testcase_37 AC 363 ms
18,228 KB
testcase_38 AC 370 ms
18,104 KB
testcase_39 AC 364 ms
18,100 KB
testcase_40 AC 2 ms
5,248 KB
testcase_41 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
//#define int long long
#define REP(i, n) for (int i = 0; i < (int)n; ++i)
#define RREP(i, n) for (int i = (int)n - 1; i >= 0; --i)
#define FOR(i, s, n) for (int i = s; i < (int)n; ++i)
#define RFOR(i, s, n) for (int i = (int)n - 1; i >= s; --i)
#define ALL(a) a.begin(), a.end()
#define IN(a, x, b) (a <= x && x < b)
template<class T>inline void out(T t){cout << t << "\n";}
template<class T,class... Ts>inline void out(T t,Ts... ts){cout << t << " ";out(ts...);}
template<class T>inline bool CHMIN(T&a,T b){if(a > b){a = b;return true;}return false;}
template<class T>inline bool CHMAX(T&a,T b){if(a < b){a = b;return true;}return false;}
constexpr int INF = 1e9;

template <typename T>
struct SegmentTree{
	using F = function<T(T,T)>;
	int n;
	F f;
	T ti;
	vector<T> dat;
	SegmentTree(){};
	SegmentTree(F f,T ti):f(f),ti(ti){}
	void init(int n_){
		n=1;
		while(n<n_) n<<=1;
		dat.assign(n<<1,ti);
	}
	void build(const vector<T> &v){
		int n_=v.size();
		init(n_);
		for(int i=0;i<n_;i++) dat[n+i]=v[i];
		for(int i=n-1;i;i--)
			dat[i]=f(dat[(i<<1)|0],dat[(i<<1)|1]);
	}
	void set_val(int k,T x){
		dat[k+=n]=x;
		while(k>>=1)
			dat[k]=f(dat[(k<<1)|0],dat[(k<<1)|1]);
	}
	T query(int a,int b){
		T vl=ti,vr=ti;
		for(int l=a+n,r=b+n;l<r;l>>=1,r>>=1) {
			if(l&1) vl=f(vl,dat[l++]);
			if(r&1) vr=f(dat[--r],vr);
		}
		return f(vl,vr);
	}
};
//auto f = [](){};
//auto seg = SegmentTree<T,decltype(f)>(f,Ti);

template< typename T >
struct edge {
  int src, to;
  T cost;

  edge(int to, T cost) : src(-1), to(to), cost(cost) {}

  edge(int src, int to, T cost) : src(src), to(to), cost(cost) {}

  edge &operator=(const int &x) {
	to = x;
	return *this;
  }

  operator int() const { return to; }
};
template< typename T >
using Edges = vector< edge< T > >;
template< typename T >
using WeightedGraph = vector< Edges< T > >;
using UnWeightedGraph = vector< vector< int > >;
template< typename T >
using Matrix = vector< vector< T > >;
template< typename G >
struct DoublingLowestCommonAncestor {
  const int LOG;
  vector< int > dep;
  const G &g;
  vector< vector< int > > table;

  DoublingLowestCommonAncestor(const G &g) : g(g), dep(g.size()), LOG(32 - __builtin_clz(g.size())) {
	table.assign(LOG, vector< int >(g.size(), -1));
  }

  void dfs(int idx, int par, int d) {
	table[0][idx] = par;
	dep[idx] = d;
	for(auto &to : g[idx]) {
	  if(to != par) dfs(to, idx, d + 1);
	}
  }

  void build() {
	dfs(0, -1, 0);// ここを書き換えれば親を選択できる
	for(int k = 0; k + 1 < LOG; k++) {
	  for(int i = 0; i < table[k].size(); i++) {
		if(table[k][i] == -1) table[k + 1][i] = -1;
		else table[k + 1][i] = table[k][table[k][i]];
	  }
	}
  }
  
  int getDep(int x){return dep[x];};
  int getDist(int a,int b){
	int parDepth = getDep(query(a,b));
	return getDep(a)-parDepth+getDep(b)-parDepth;
  }

  int query(int u, int v) {
	if(dep[u] > dep[v]) swap(u, v);
	for(int i = LOG - 1; i >= 0; i--) {
	  if(((dep[v] - dep[u]) >> i) & 1) v = table[i][v];
	}
	if(u == v) return u;
	for(int i = LOG - 1; i >= 0; i--) {
	  if(table[i][u] != table[i][v]) {
		u = table[i][u];
		v = table[i][v];
	  }
	}
	return table[0][u];
  }
};


signed main(){
	int N,K,Q;
	cin >> N >> K >> Q;
	vector<int>c(N);
	REP(i,N){
		cin >> c[i];
	}
	vector<int>a(K);
	REP(i,K){
		cin >> a[i];
		--a[i];
	}
	vector<vector<int>>g(N);
	REP(i,N - 1){
		int a,b;
		cin >> a >> b;
		--a;--b;
		g[a].emplace_back(b);
		g[b].emplace_back(a);
	}
	auto dfs = [&](auto&&f,int now,int par)->void{
		if(par != -1)CHMAX(c[now],c[par]);
		for(auto e:g[now]){
			if(e == par)continue;
			f(f,e,now);
		}
	};
	dfs(dfs,0,-1);
	DoublingLowestCommonAncestor<vector<vector<int>>>lca(g);
	lca.build();
	SegmentTree<int>seg([&](int a,int b){
		if(a == -1)return b;
		if(b == -1)return a;
		return lca.query(a,b);
	},-1);
	seg.build(a);
	vector<int>ans;
	while(Q--){
		int t;
		cin >> t;
		if(t == 1){
			int x,y;
			cin >> x >> y;
			--x;--y;
			seg.set_val(x,y);
		}else{
			int l,r;
			cin >> l >> r;
			--l;--r;
			ans.emplace_back(seg.query(l,r + 1));
		}
	}
	for(auto e:ans)out(c[e]);
}
0