全WA求调
查看原帖
全WA求调
1079493
Hy13_xsm楼主2024/7/12 13:51
#include<bits/stdc++.h>
#define int long long
#define N 100005
using namespace std;
int n,m,cnt,in[N],f[N],sz[N],son[N],d[N],top[N],rnk[N],dfn[N];
vector<int>g[N];
struct node{
	int l,r,sum,top,lz;
}seg[4*N];
int ls(int p){
	return 2*p;
}
int rs(int p){
	return 2*p+1;
}
void push_down(int p)
{
	seg[ls(p)].lz+=seg[p].lz;
	seg[rs(p)].lz+=seg[p].lz;
	seg[ls(p)].sum+=(seg[ls(p)].r-seg[ls(p)].l+1)*seg[p].lz;
	seg[rs(p)].sum+=(seg[rs(p)].r-seg[rs(p)].l+1)*seg[p].lz;
	seg[p].lz=0;
}
void dfs1(int u,int fa)
{
	f[u]=fa;
	d[u]=d[fa]+1;
	sz[u]=1;
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(v==fa)continue;
		dfs1(v,u);
		sz[u]+=sz[v];
		if(sz[son[u]]<sz[v])son[u]=v;
	}
}
void dfs2(int u,int _top)
{
	top[u]=_top;
	dfn[u]=++cnt;
	rnk[cnt]=u;
	if(son[u])dfs2(son[u],_top);
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(v==f[u]||v==son[u])continue;
		dfs2(v,v);
	}
}
void build(int p,int l,int r)
{
	seg[p].l=l;seg[p].r=r;
	if(l==r){
		seg[p].top=in[rnk[l]];
		seg[p].sum=in[rnk[l]];
		return;
	}
	int mid=(l+r)>>1;
	build(ls(p),l,mid);
	build(rs(p),mid+1,r);
	seg[p].sum=seg[ls(p)].sum+seg[rs(p)].sum;
	seg[p].top=max(seg[ls(p)].top,seg[rs(p)].top);
}
void update(int p,int l,int r,int k)
{
	if(seg[p].l>r||seg[p].r<l)return;
	if(seg[p].l>=l&&seg[p].r<=r){
		seg[p].lz+=k;
		seg[p].sum+=k;
		seg[p].top+=k;
		return;
	}
	push_down(p);
	update(ls(p),l,r,k);
	update(rs(p),l,r,k);
	seg[p].sum=seg[ls(p)].sum+seg[rs(p)].sum;
	seg[p].top=max(seg[ls(p)].top,seg[rs(p)].top);
}
int query_max(int p,int l,int r)
{
	if(seg[p].l>r||seg[p].r<l)return INT_MIN;
	if(seg[p].l>=l&&seg[p].r<=r)return seg[p].top;
	push_down(p);
	return max(query_max(ls(p),l,r),query_max(rs(p),l,r));
}
int query_sum(int p,int l,int r)
{
	if(seg[p].l>r||seg[p].r<l)return 0;
	if(seg[p].l>=l&&seg[p].r<=r)return seg[p].sum;
	push_down(p);
	return query_sum(ls(p),l,r)+query_sum(rs(p),l,r);
}
int get_max(int x,int y)
{
	int ans=INT_MIN;
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		ans=max(ans,query_max(1,dfn[top[x]],dfn[x]));
		x=f[top[x]];
	}
	if(d[x]>d[y])swap(x,y);
	return max(ans,query_max(1,dfn[x],dfn[y]));
}
int get_sum(int x,int y)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		ans+=query_sum(1,dfn[top[x]],dfn[x]);
		x=f[top[x]];
	}
	if(d[x]>d[y])swap(x,y);
	return ans+query_sum(1,dfn[x],dfn[y]);
}
signed main()
{
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	for(int i=1;i<=n;i++)
	cin>>in[i];
	dfs1(1,0);
	dfs2(1,1);
	build(1,1,n);
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		string op;
		int x,y;
		cin>>op>>x>>y;
		if(op=="CHANGE")update(1,dfn[x],dfn[x],y-in[x]);
		else if(op=="QMAX")cout<<get_max(x,y)<<endl;
		else cout<<get_sum(x,y)<<endl;
	}
	return 0;
}
2024/7/12 13:51
加载中...