求调
查看原帖
求调
428449
Amon_Xolotl楼主2023/6/27 08:51
#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int n,q;
int h[N],e[N<<1],ne[N],cnt;
int size[N],top[N],dep[N],son[N],id[N],fa[N],len,w[N],c[N];
int root[N<<2];
void add(int a,int b)
{
	e[++cnt]=b,ne[cnt]=h[a],h[a]=cnt;
}
int rt;
void dfs1(int u,int ba)
{
	size[u]=1;
	for(int i=h[u];i;i=ne[i])
	{
		int v=e[i];
		if(v!=ba)
		{
			fa[v]=u;
			dep[v]=dep[u]+1;
			dfs1(v,u);
			size[u]+=size[v];
			if(size[son[u]]<size[v])
			{
				son[u]=v;
			}
		}
	}
}
void dfs2(int u)
{
	if(!u)
	{
		return;
	}
	id[u]=++cnt;
	if(son[u])
	{
		top[son[u]]=top[u];
		dfs2(son[u]);
	}
	for(int i=h[u];i;i=ne[i])
	{
		int v=e[i];
		if(v!=son[u]&&v!=fa[u])
		{
			top[v]=v;
			dfs2(v);
		}
	}
}
struct zjy
{
	int l,r,max,sum;
}tree[N<<2];
void update(int &now,int w,int l,int r,int x)
{
	if(!now)
	{
		now=++len;
	}
	tree[now].max=max(tree[now].max,w),tree[now].sum+=w;
	if(l==r)
	{
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)
	{
		update(tree[now].l,w,l,mid,x);
	}
	else
	{
		update(tree[now].r,w,mid+1,r,x);
	}
}
void remove(int &now,int l,int r,int x)
{
	if(l==r)
	{
		tree[now].max=0,tree[now].max=0;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)
	{
		remove(tree[now].l,l,mid,x);
	}
	else
	{
		remove(tree[now].r,mid+1,r,x);
	}
	tree[now].sum=tree[tree[now].l].sum+tree[tree[now].r].sum;
	tree[now].max=max(tree[tree[now].l].max,tree[tree[now].r].max);
}
int querysum1(int now,int l,int r,int x,int y)
{
	if(r<x||l>y)
	{
		return 0;
	}
	if(l>=x&&r<=y)
	{
		return tree[now].sum;
	}
	int mid=(l+r)>>1;
	return querysum1(tree[now].l,l,mid,x,y)+querysum1(tree[now].r,mid+1,r,x,y);
}
int querymax1(int now,int l,int r,int x,int y)
{
	if(r<x||l>y)
	{
		return 0;
	}
	if(l>=x&&r<=y)
	{
		return tree[now].max;
	}
	int mid=(l+r)>>1;
	return max(querymax1(tree[now].l,l,mid,x,y),querymax1(tree[now].r,mid+1,r,x,y));
}
int querysum2(int x,int y,int zj)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])
		{
			swap(x,y);
		}
		ans+=querysum1(root[zj],1,n,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])
	{
		swap(x,y);
	}
	ans+=querysum1(root[zj],1,n,id[y],id[x]);
	return ans;
}
int querymax2(int x,int y,int zj)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])
		{
			swap(x,y);
		}
		ans=max(querymax1(root[zj],1,n,id[top[x]],id[x]),ans);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])
	{
		swap(x,y);
	}
	ans=max(ans,querymax1(root[zj],1,n,id[y],id[x]));
	return ans;
}
int main()
{
	len=0;
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;++i)
	{
		scanf("%d%d",&w[i],&c[i]);
	}
	for(int i=1;i<=n;++i)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	fa[1]=0,dep[1]=1;dfs1(1,0);
	cnt=0;
	top[1]=1,dfs2(1);
	for(int i=1;i<=n;++i)
	{
		update(root[c[i]],w[i],1,n,id[i]);
	}
	char s[10];
	while(q--)
	{
		int x,y;
		cin>>s>>x>>y;
		if(s[1]=='C')
		{
			remove(root[c[x]],1,n,id[x]);
			update(root[y],w[x],1,n,id[x]);
			c[x]=y;
		}
		else if(s[1]=='W')
		{
			remove(root[c[x]],1,n,id[x]);
			update(root[c[x]],y,1,n,id[x]);
			w[x]=y;
		}
		else if(s[1]=='S')
		{
			printf("%d\n",querysum2(x,y,c[x]));
		}
		else if(s[1]=='M')
		{
			printf("%d\n",querymax2(x,y,c[x]));
		}
	}
	return 0;
}
2023/6/27 08:51
加载中...