Cod sursa(job #3325520)

Utilizator MagicantPlusIuoras Andrei MagicantPlus Data 25 noiembrie 2025 17:20:07
Problema Heavy Path Decomposition Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 13.37 kb
// #include <bits/stdc++.h>

// using namespace std;

// #define ll int
// #define ull unsigned ll
// #define vec vector
// #define flist forward_list
// #define um unordered_map
// #define us unordered_set
// #define prioq priority_queue
// #define all(x) x.begin(), x.end()
// #define pb push_back
// #define pf push_front
// #define fi first
// #define se second
// #define ld long double
// #define cd complex<ld>
// #define pll pair<ll, ll>
// #define tll tuple<ll, ll, ll>
// #define ass assign
// #define fun function

// #define LL_MAX LLONG_MAX
// #define LL_MIN LLONG_MIN
// #define ULL_MAX ULLONG_MAX
// #define LD_MAX LDBL_MAX
// #define LD_MIN LDBL_MIN

// #define nl '\n'

// #define vv(type, name, n, ...) vector<vector<type>> name(n, vector<type>(__VA_ARGS__))
// #define vvv(type, name, n, m, ...) \
//     vector<vector<vector<type>>> name(n, vector<vector<type>>(m, vector<type>(__VA_ARGS__)))

// // https://trap.jp/post/1224/
// #define rep1(n) for(ll i=0; i<(ll)(n); ++i)
// #define rep2(i,n) for(ll i=0; i<(ll)(n); ++i)
// #define rep3(i,a,b) for(ll i=(ll)(a); i<(ll)(b); ++i)
// #define rep4(i,a,b,c) for(ll i=(ll)(a); i<(ll)(b); i+=(ll)(c))
// #define cut4(a,b,c,d,e,...) e
// #define rep(...) cut4(__VA_ARGS__,rep4,rep3,rep2,rep1)(__VA_ARGS__)
// #define rep1e(n) for(ll i=1; i<=(ll)(n); ++i)
// #define rep2e(i,n) for(ll i=1; i<=(ll)(n); ++i)
// #define rep3e(i,a,b) for(ll i=(ll)(a); i<=(ll)(b); ++i)
// #define rep4e(i,a,b,c) for(ll i=(ll)(a); i<=(ll)(b); i+=(ll)(c))
// #define repe(...) cut4(__VA_ARGS__,rep4e,rep3e,rep2e,rep1e)(__VA_ARGS__)
// #define per1(n) for(ll i=(ll)(n)-1; i>=0; --i)
// #define per2(i,n) for(ll i=(ll)(n)-1; i>=0; --i)
// #define per3(i,a,b) for(ll i=(ll)(a)-1; i>=(ll)(b); --i)
// #define per4(i,a,b,c) for(ll i=(ll)(a)-1; i>=(ll)(b); i-=(ll)(c))
// #define per(...) cut4(__VA_ARGS__,per4,per3,per2,per1)(__VA_ARGS__)
// #define per1e(n) for(ll i=(ll)(n); i>=1; --i)
// #define per2e(i,n) for(ll i=(ll)(n); i>=1; --i)
// #define per3e(i,a,b) for(ll i=(ll)(a); i>=(ll)(b); --i)
// #define per4e(i,a,b,c) for(ll i=(ll)(a); i>=(ll)(b); i-=(ll)(c))
// #define pere(...) cut4(__VA_ARGS__,per4e,per3e,per2e,per1e)(__VA_ARGS__)
// #define rep_subset(i,s) for(ll i=0;i!=-1;i=i==(s)?-1:(i|~(s))+1&(s))
// #define per_subset(i,s) for(ll i=(s);i!=-1;i=i==0?-1:i-1&(s))
// // #define forit(v) for (auto it = v.begin(); it != v.end(); it++)
// // #define forit2(it, v) for (auto it = v.begin(); it != v.end(); it++)
// // #define forrit(v) for (auto it = v.rbegin(); it != v.rend(); it++)
// // #define forrit2(it, v) for (auto it = v.rbegin(); it != v.rend(); it++)

// // #ifdef i_am_noob
// #define bug(...) cerr << "#" << __LINE__ << ' ' << #__VA_ARGS__ << "- ", _do(__VA_ARGS__)
// template<typename T> void _do(vector<vector<T>> x){cerr << "\n"; for(auto i: x) {for(auto j : i) {cerr << j << ' ';} cerr << "\n";}}
// template<typename T> void _do(vector<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
// template<typename T> void _do(set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
// template<typename T> void _do(unordered_set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
// template<typename T> void _do(T && x) {cerr << x << endl;}
// template<typename T, typename ...S> void _do(T && x, S&&...y) {cerr << x << ", "; _do(y...);}
// // #else
// // #define bug(...) 777771449
// // #endif
// template<class T, class S> bool mkmin(T& a, const initializer_list<S>& b)
// {S m = *min_element(all(b)); if(m<a){a=m;return 1;}return 0;}
// template<class T, class S> bool mkmax(T& a, const initializer_list<S>& b)
// {S m = *max_element(all(b)); if(m>a){a=m;return 1;}return 0;}
// template<class T, class S> bool mkmin(T& a, const S& b){if(b<a){a=b;return 1;}return 0;}
// template<class T, class S> bool mkmax(T& a, const S& b){if(b>a){a=b;return 1;}return 0;}

// const ll MOD = 1e9 + 7;
// const ll INF = (1ll << 27);

// const string FILENAME = "heavypath";
// ifstream fin(FILENAME + ".in");
// ofstream fout(FILENAME + ".out");

// #define cin fin
// #define cout fout

// class HLD;
// class Hoshino
// {
// public:

// 	ll n;
// 	vec<ll> ai;
	
// 	pll calc(ll i, ll l, ll r){return{(l+r)/2, i+2*((l+r)/2-l+1)};}
// 	void build(vec<ll>& nums)
// 	{
// 		n=nums.size();
// 		ai.resize(2*n-1);
		
// 		fun<void(ll,ll,ll)>build_([&](ll i, ll l, ll r)
// 		{
// 			if(l==r)
// 			{
// 				ai[i] = nums[l];
// 				return;
// 			}
			
// 			ll mid,rc;tie(mid,rc)=calc(i,l,r);
// 			build_(i+1,l,mid);
// 			build_(rc,mid+1,r);
			
// 			ai[i] = max(ai[i+1],ai[rc]);
// 		});
// 		build_(0,0,n-1);
// 	}
// 	void update(ll x, ll val)
// 	{
// 		fun<void(ll,ll,ll)>update_([&](ll i, ll l, ll r)
// 		{
// 			if(l==r)
// 			{
// 				ai[i] = val;
// 				return;
// 			}
			
// 			ll mid,rc;tie(mid,rc)=calc(i,l,r);
// 			if(x<=mid)update_(i+1,l,mid);
// 			else update_(rc,mid+1,r);
			
// 			ai[i] = max(ai[i+1],ai[rc]);
// 		});
// 		update_(0,0,n-1);
// 	}
// 	ll query(ll x, ll y)
// 	{
// 		ll ans = -INF;
		
// 		fun<void(ll,ll,ll)>query_([&](ll i, ll l, ll r)
// 		{
// 			if(x<=l&&r<=y)
// 			{
// 				mkmax(ans,ai[i]);
// 				return;
// 			}
			
// 			ll mid,rc;tie(mid,rc)=calc(i,l,r);
// 			if(x<=mid)query_(i+1,l,mid);
// 			if(mid<y)query_(rc,mid+1,r);
// 		});
// 		query_(0,0,n-1);
		
// 		return ans;
// 	}
// };

// class Tree
// {
// public:

// 	ll n;
// 	vec<flist<ll>> adj;
// 	vec<ll> cost;
// 	HLD *hld;
	
// 	void read();
// };

// class HLD
// {
// public:

// 	vec<ll> p, d, v, pos, h, hvy;
// 	Hoshino ai;
	
// 	void build(Tree *t)
// 	{
// 		ll n=t->n;
// 		p.resize(n);
// 		d.resize(n);
// 		v.reserve(n);
// 		pos.resize(n);
// 		h.resize(n);
// 		hvy.resize(n);
		
// 		fun<ll(ll)>dfs([&](ll o)->ll
// 		{
// 			ll size=1, maxsize=-1;
			
// 			for(auto u : t->adj[o])if(u!=p[o])
// 			{
// 				p[u]=o;
// 				d[u]=d[o]+1;
// 				ll usize = dfs(u);
// 				if(mkmax(maxsize,usize)) hvy[o] = u;
// 				size += usize;
// 			}
			
// 			return size;
// 		});
// 		p[0]=-1;
// 		d[0]=0;
// 		dfs(0);
		
// 		fun<void(ll)>dfs2([&](ll o)
// 		{
// 			v.pb(t->cost[o]);
// 			pos[o] = v.size()-1;
			
// 			if(hvy[o]!=-1) h[hvy[o]] = h[o], dfs2(hvy[o]);
			
// 			for(auto u : t->adj[o])if(u!=p[o] && u!=hvy[o]) h[u] = u, dfs2(u);
// 		});
//         h[0] = 0;
// 		dfs2(0);
		
// 		ai.build(v);
// 	}
// 	void update(ll node, ll val)
// 	{
// 		ai.update(pos[node], val);
// 	}
// 	ll query(ll a, ll b)
// 	{
// 		ll ans=-1;
// 		while(h[a]!=h[b])
// 		{
// 			if(d[h[a]]<d[h[b]])swap(a,b);
// 			mkmax(ans,ai.query(pos[h[a]],pos[a]));
// 			a=p[h[a]];
// 		}
// 		if(d[a]<d[b])swap(a,b);
// 		return max(ans, ai.query(pos[b],pos[a]));
// 	}
// };

// void Tree::read()
// {
// 	ll q;
// 	cin >> n >> q;
// 	adj.resize(n);
// 	cost.resize(n);
// 	rep(n) cin >> cost[i];
// 	rep(n-1)
// 	{
// 		ll a,b;
// 		cin >> a >> b;
// 		a--,b--;
// 		adj[a].pf(b);
// 		adj[b].pf(a);
// 	}
	
// 	hld = new HLD;
// 	hld->build(this);
	
// 	rep(q)
// 	{
// 		ll t, x, y;
// 		cin >> t >> x >> y;
		
// 		if(t==0)
// 		{
// 			x--;
// 			hld->update(x, y);
// 		}
// 		else
// 		{
// 			x--,y--;
// 			cout << hld->query(x,y) << nl;
// 		}
// 	}
// }

// int main()
// {
// 	Tree t;
// 	t.read();
	
//     return 0;
// }
#include <bits/stdc++.h>

using namespace std;

#define ll int
#define ull unsigned ll
#define vec vector
#define flist forward_list
#define um unordered_map
#define us unordered_set
#define prioq priority_queue
#define all(x) x.begin(), x.end()
#define pb push_back
#define pf push_front
#define fi first
#define se second
#define ld long double
#define cd complex<ld>
#define pll pair<ll, ll>
#define tll tuple<ll, ll, ll>
#define ass assign
#define fun function

#define LL_MAX LLONG_MAX
#define LL_MIN LLONG_MIN
#define ULL_MAX ULLONG_MAX
#define LD_MAX LDBL_MAX
#define LD_MIN LDBL_MIN

#define nl '\n'

#define vv(type, name, n, ...) vector<vector<type>> name(n, vector<type>(__VA_ARGS__))
#define vvv(type, name, n, m, ...) \
    vector<vector<vector<type>>> name(n, vector<vector<type>>(m, vector<type>(__VA_ARGS__)))

// https://trap.jp/post/1224/
#define rep1(n) for(ll i=0; i<(ll)(n); ++i)
#define rep2(i,n) for(ll i=0; i<(ll)(n); ++i)
#define rep3(i,a,b) for(ll i=(ll)(a); i<(ll)(b); ++i)
#define rep4(i,a,b,c) for(ll i=(ll)(a); i<(ll)(b); i+=(ll)(c))
#define cut4(a,b,c,d,e,...) e
#define rep(...) cut4(__VA_ARGS__,rep4,rep3,rep2,rep1)(__VA_ARGS__)
#define rep1e(n) for(ll i=1; i<=(ll)(n); ++i)
#define rep2e(i,n) for(ll i=1; i<=(ll)(n); ++i)
#define rep3e(i,a,b) for(ll i=(ll)(a); i<=(ll)(b); ++i)
#define rep4e(i,a,b,c) for(ll i=(ll)(a); i<=(ll)(b); i+=(ll)(c))
#define repe(...) cut4(__VA_ARGS__,rep4e,rep3e,rep2e,rep1e)(__VA_ARGS__)
#define per1(n) for(ll i=(ll)(n)-1; i>=0; --i)
#define per2(i,n) for(ll i=(ll)(n)-1; i>=0; --i)
#define per3(i,a,b) for(ll i=(ll)(a)-1; i>=(ll)(b); --i)
#define per4(i,a,b,c) for(ll i=(ll)(a)-1; i>=(ll)(b); i-=(ll)(c))
#define per(...) cut4(__VA_ARGS__,per4,per3,per2,per1)(__VA_ARGS__)
#define per1e(n) for(ll i=(ll)(n); i>=1; --i)
#define per2e(i,n) for(ll i=(ll)(n); i>=1; --i)
#define per3e(i,a,b) for(ll i=(ll)(a); i>=(ll)(b); --i)
#define per4e(i,a,b,c) for(ll i=(ll)(a); i>=(ll)(b); i-=(ll)(c))
#define pere(...) cut4(__VA_ARGS__,per4e,per3e,per2e,per1e)(__VA_ARGS__)
#define rep_subset(i,s) for(ll i=0;i!=-1;i=i==(s)?-1:(i|~(s))+1&(s))
#define per_subset(i,s) for(ll i=(s);i!=-1;i=i==0?-1:i-1&(s))
// #define forit(v) for (auto it = v.begin(); it != v.end(); it++)
// #define forit2(it, v) for (auto it = v.begin(); it != v.end(); it++)
// #define forrit(v) for (auto it = v.rbegin(); it != v.rend(); it++)
// #define forrit2(it, v) for (auto it = v.rbegin(); it != v.rend(); it++)

// #ifdef i_am_noob
#define bug(...) cerr << "#" << __LINE__ << ' ' << #__VA_ARGS__ << "- ", _do(__VA_ARGS__)
template<typename T> void _do(vector<vector<T>> x){cerr << "\n"; for(auto i: x) {for(auto j : i) {cerr << j << ' ';} cerr << "\n";}}
template<typename T> void _do(vector<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(unordered_set<T> x){for(auto i: x) cerr << i << ' ';cerr << "\n";}
template<typename T> void _do(T && x) {cerr << x << endl;}
template<typename T, typename ...S> void _do(T && x, S&&...y) {cerr << x << ", "; _do(y...);}
// #else
// #define bug(...) 777771449
// #endif
template<class T, class S> bool mkmin(T& a, const initializer_list<S>& b)
{S m = *min_element(all(b)); if(m<a){a=m;return 1;}return 0;}
template<class T, class S> bool mkmax(T& a, const initializer_list<S>& b)
{S m = *max_element(all(b)); if(m>a){a=m;return 1;}return 0;}
template<class T, class S> bool mkmin(T& a, const S& b){if(b<a){a=b;return 1;}return 0;}
template<class T, class S> bool mkmax(T& a, const S& b){if(b>a){a=b;return 1;}return 0;}

const ll MOD = 1e9 + 7;
const ll INF = (1ll << 27);

const string FILENAME = "heavypath";
ifstream fin(FILENAME + ".in");
ofstream fout(FILENAME + ".out");

#define cin fin
#define cout fout

class HLD;
class Hoshino
{
public:

	ll n;
	vec<ll> ai;
	
	pll calc(ll i, ll l, ll r){return{(l+r)/2, i+2*((l+r)/2-l+1)};}
	void build(vec<ll>& nums)
	{
		n=nums.size();
		ai.resize(2*n-1);
		
		fun<void(ll,ll,ll)>build_([&](ll i, ll l, ll r)
		{
			if(l==r)
			{
				ai[i] = nums[l];
				return;
			}
			
			ll mid,rc;tie(mid,rc)=calc(i,l,r);
			build_(i+1,l,mid);
			build_(rc,mid+1,r);
			
			ai[i] = max(ai[i+1],ai[rc]);
		});
		build_(0,0,n-1);
	}
	void update(ll x, ll val)
	{
		fun<void(ll,ll,ll)>update_([&](ll i, ll l, ll r)
		{
			if(l==r)
			{
				ai[i] = val;
				return;
			}
			
			ll mid,rc;tie(mid,rc)=calc(i,l,r);
			if(x<=mid)update_(i+1,l,mid);
			else update_(rc,mid+1,r);
			
			ai[i] = max(ai[i+1],ai[rc]);
		});
		update_(0,0,n-1);
	}
	ll query(ll x, ll y)
	{
		ll ans = -INF;
		
		fun<void(ll,ll,ll)>query_([&](ll i, ll l, ll r)
		{
			if(x<=l&&r<=y)
			{
				mkmax(ans,ai[i]);
				return;
			}
			
			ll mid,rc;tie(mid,rc)=calc(i,l,r);
			if(x<=mid)query_(i+1,l,mid);
			if(mid<y)query_(rc,mid+1,r);
		});
		query_(0,0,n-1);
		
		return ans;
	}
};

class Tree
{
public:

	ll n;
	vec<flist<ll>> adj;
	vec<ll> cost;
	HLD *hld;
	
	void read();
};

class HLD
{
public:

	vec<ll> p, d, v, pos, head, hvy;
	Hoshino ai;
	
	void build(Tree *t)
	{
		ll n=t->n;
		p.resize(n);
		d.resize(n);
		v.reserve(n);
		pos.resize(n);
		head.resize(n);
		hvy.resize(n);
		
		fun<ll(ll)>dfs([&](ll o)->ll
		{
			ll size=1, maxsize=-1, maxu=-1;
			
			for(auto u : t->adj[o])if(u!=p[o])
			{
				p[u]=o;
				d[u]=d[o]+1;
				ll usize = dfs(u);
				if(mkmax(maxsize,usize)) maxu=u;
				size += usize;
			}
			
			hvy[o]=maxu;
			return size;
		});
		p[0]=-1;
		d[0]=0;
		dfs(0);
		
		fun<void(ll,ll)>dfs2([&](ll o, ll h)
		{
			head[o] = h;
			v.pb(t->cost[o]);
			pos[o] = v.size()-1;
			
			if(hvy[o]!=-1) dfs2(hvy[o],h);
			
			for(auto u : t->adj[o])if(u!=p[o] && u!=hvy[o]) dfs2(u,u);
		});
		dfs2(0,0);
		
		ai.build(v);
	}
	void update(ll node, ll val)
	{
		ai.update(pos[node], val);
	}
	ll query(ll a, ll b)
	{
		ll ans=-1;
		while(head[a]!=head[b])
		{
			if(d[head[a]]<d[head[b]])swap(a,b);
			mkmax(ans,ai.query(pos[head[a]],pos[a]));
			a=p[head[a]];
		}
		if(d[a]<d[b])swap(a,b);
		return max(ans, ai.query(pos[b],pos[a]));
	}
};

void Tree::read()
{
	ll q;
	cin >> n >> q;
	adj.resize(n);
	cost.resize(n);
	rep(n) cin >> cost[i];
	rep(n-1)
	{
		ll a,b;
		cin >> a >> b;
		a--,b--;
		adj[a].pf(b);
		adj[b].pf(a);
	}
	
	hld = new HLD;
	hld->build(this);
	
	rep(q)
	{
		ll t, x, y;
		cin >> t >> x >> y;
		
		if(t==0)
		{
			x--;
			hld->update(x, y);
		}
		else
		{
			x--,y--;
			cout << hld->query(x,y) << nl;
		}
	}
}

int main()
{
	Tree t;
	t.read();
	
    return 0;
}