萌新求助,只能过样例!
查看原帖
萌新求助,只能过样例!
409394
AssassinQ楼主2023/6/19 14:04
#include<bits/stdc++.h>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const ll MAXN=3e4+5;
ll n,a,b,w[MAXN],rnk[MAXN],cnt,T;
vector<ll> v[MAXN];
string str;
struct point{
	ll fa,dep,siz,son,top,dfn;
}p[MAXN];
struct segmenttree{
	ll l,r,maxd,sumd;
}s[MAXN<<2];
void dfs1(ll x){
	ll t=0;
	p[x].siz=1;
	for(int i=0;i<v[x].size();i++)
		if(!p[v[x][i]].dep){
			t=v[x][i];
			p[t].dep=p[x].dep+1;
			p[t].fa=x;
			dfs1(t);
			p[x].siz+=p[t].siz;
			if(!p[x].son||p[t].siz>p[p[x].son].siz) p[x].son=t;
		}
	return;
}
void dfs2(ll x,ll f){
	p[x].top=f;
	p[x].dfn=++cnt;
	rnk[cnt]=x;
	if(!p[x].son) return;
	dfs2(p[x].son,f);
	for(int i=0;i<v[x].size();i++)
		if(v[x][i]!=p[x].son&&v[x][i]!=p[x].fa)
			dfs2(v[x][i],v[x][i]); 
	return;
}
inline void push_up(ll t){
	s[t].maxd=max(s[t<<1].maxd,s[t<<1|1].maxd);
	s[t].sumd=s[t<<1].sumd+s[t<<1|1].sumd;
	return;
}
void build(ll t,ll l,ll r){
	ll mid=(l+r)>>1;
	s[t].l=l; s[t].r=r;
	if(l==r){
		s[t].maxd=s[t].sumd=w[rnk[l]];
		return;
	}
	build(t<<1,l,mid);
	build(t<<1|1,mid+1,r);
	push_up(t);
	return;
}
void change(ll t,ll x,ll y){
	ll mid=(s[t].l+s[t].r)>>1;
	if(s[t].l==s[t].r){
		s[t].maxd=max(s[t].maxd,y);
		s[t].sumd=y;
		return;
	}
	if(x<=mid) change(t<<1,x,y);
	else change(t<<1|1,x,y);
	push_up(t);
	return;
}
ll ask(ll t,ll l,ll r,bool emax){
	ll mid=(s[t].l+s[t].r)>>1,t1=0,t2=0;
	if(l<=s[t].l&&s[t].r<=r)
		return emax?s[t].maxd:s[t].sumd;
	if(l<=mid) t1=ask(t<<1,l,r,emax);
	if(r>mid) t2=ask(t<<1|1,l,r,emax);
	if(emax) return max(t1,t2);
	return t1+t2;
}
ll answer(ll x,ll y,bool emax){
	ll fx=x,fy=y,t=0,ans=emax?-inf:0;
	while(p[fx].top!=p[fy].top){
		if(p[p[fx].top].dep>p[p[fy].top].dep){
			t=fx;
			fx=p[p[fx].top].fa;
		}
		else{
			t=fy;
			fy=p[p[fy].top].fa;
		}
		if(emax) ans=max(ans,ask(1,p[p[t].top].dfn,p[t].dfn,true));
		else ans+=ask(1,p[p[t].top].dfn,p[t].dfn,false);
	}
	fx=p[fx].dfn,fy=p[fy].dfn;
	if(fx>fy) swap(fx,fy);
	if(emax) return max(ans,ask(1,fx,fy,true));
	return ans+ask(1,fx,fy,false);
}
int main(){
	scanf("%lld",&n);
	for(int i=1;i<n;i++){
		scanf("%lld%lld",&a,&b);
		v[a].push_back(b);
		v[b].push_back(a);
	}
	for(int i=1;i<=n;i++)
		scanf("%lld",&w[i]);
	p[1].dep=1;
	dfs1(1);
	dfs2(1,1);
	build(1,1,n);
	scanf("%lld",&T);
	while(T--){
		cin>>str;
		scanf("%lld%lld",&a,&b);
		if(str=="QMAX") printf("%lld\n",answer(a,b,true));
		else if(str=="QSUM") printf("%lld\n",answer(a,b,false));
		else change(1,p[a].dfn,b);
	}
	return 0;
}
2023/6/19 14:04
加载中...