洛谷P3313 旅行

题目描述

S国有N个城市，编号从1到N。城市间用N-1条双向道路连接，满足从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教，如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。

为了方便，我们用不同的正整数代表各种宗教， S国的居民常常旅行。旅行时他们总会走最短路，并且为了避免麻烦，只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级，旅行者们常会记下途中（包括起点和终点）留宿过的城市的评级总和或最大值。

在S国的历史上常会发生以下几种事件：

“CC x c“：城市x的居民全体改信了c教；

“CW x w“：城市x的评级调整为w;

“QS x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级总和；

“QM x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级最大值。

由于年代久远，旅行者记下的数字已经遗失了，但记录开始之前每座城市的信仰与评级，还有事件记录本身是完好的。请根据这些信息，还原旅行者记下的数字。 为了方便，我们认为事件之间的间隔足够长，以致在任意一次旅行中，所有城市的评级和信仰保持不变。

解析：

1.有多少个宗教，就建多少颗线段树

2.线段树要用动态开点，建树的方式很像主席树，合并的方式像线段树。

3.修改操作1就是先删掉x是宗教线段树中的x城市，把x城市的宗教改为c，再向c宗教线段树中插入x城市。

4.修改操作2直接把x城市的评级修改为w就好了。这个insert函数非常神奇，在建树的时候会加点，而在修改的时候不会加点，这样就可以少写一个函数了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct tree
{
	int max;
	int sum;
	int l,r;
}tr[4000010];
struct point
{
	int next;
	int to;
}e[200001];
int n,m,q,num,cnt,x,y;
int fa[100010],head[100010],top[100010],size[100010],T[100010];
int deep[100010],id[100010],c[100010],w[100010],son[100010];
void add(int from,int to)
{
	e[++num].next=head[from];
	e[num].to=to;
	head[from]=num;
}
void dfs1(int x)
{
	size[x]=1;
	for(int i=head[x];i!=0;i=e[i].next)
	{
		int to=e[i].to;
		if(to!=fa[x])
		{
			fa[to]=x;
			deep[to]=deep[x]+1;
			dfs1(to);
			size[x]+=size[to];
			if(size[to]>size[son[x]])
				son[x]=to;
		}
	}
}
void dfs2(int x,int tp)
{
	top[x]=tp;
	id[x]=++cnt;
	if(son[x])
		dfs2(son[x],tp);
	for(int i=head[x];i!=0;i=e[i].next)
	{
		int to=e[i].to;
		if(to==fa[x]||to==son[x])
			continue;
		dfs2(to,to);
	}
}
void updata(int node)
{
	tr[node].max=max(tr[tr[node].l].max,tr[tr[node].r].max);
	tr[node].sum=tr[tr[node].l].sum+tr[tr[node].r].sum;
	return;
}
void pushdown(int node)
{
	tr[node].l=tr[node].r=tr[node].max=tr[node].sum=0;
}
void insert(int &node,int l,int r,int x,int v)
{
	if(!node)
		node=++cnt;
	if(l==r)
	{
		tr[node].max=tr[node].sum=v;
		return;
	}
	int mid=(l+r)/2;
	if(x<=mid)
		insert(tr[node].l,l,mid,x,v);
	else
		insert(tr[node].r,mid+1,r,x,v);
	updata(node);
}
void delt(int &node,int l,int r,int x)
{
	if(l==r)
	{
		pushdown(node);
		node=0;
		return;
	}
	int mid=(l+r)/2;
	if(x<=mid)
		delt(tr[node].l,l,mid,x);
	else
		delt(tr[node].r,mid+1,r,x);
	updata(node);
	if(!tr[node].l&&!tr[node].r)
	{
		pushdown(node);
		node=0;
	}
}
int query1(int node,int l,int r,int l1,int r1)
{
	if(!node)
		return 0;
	if(l1<=l&&r1>=r)
		return tr[node].sum;
	if(l1>r||r1<l)
		return 0;
	int mid=(l+r)/2;
	return query1(tr[node].l,l,mid,l1,r1)+query1(tr[node].r,mid+1,r,l1,r1);
}
int query2(int node,int l,int r,int l1,int r1)
{
	if(!node)
		return 0;
	if(l1<=l&&r1>=r)
		return tr[node].max;
	if(l1>r||r1<l)
		return 0;
	int mid=(l+r)/2;
	return max(query2(tr[node].l,l,mid,l1,r1),query2(tr[node].r,mid+1,r,l1,r1));
}
int getsum(int x,int y)
{
	int ans=0,root=c[x];
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]])
			swap(x,y);
		ans+=query1(T[root],1,n,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(deep[x]>deep[y])
		swap(x,y);
	ans+=query1(T[root],1,n,id[x],id[y]);
	return ans;
}
int getmax(int x,int y)
{
	int ans=0,root=c[x];
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]])
			swap(x,y);
		ans=max(ans,query2(T[root],1,n,id[top[x]],id[x]));
		x=fa[top[x]];
	}
	if(deep[x]>deep[y])
		swap(x,y);
	ans=max(ans,query2(T[root],1,n,id[x],id[y]));
	return ans;
}
int main()
{
	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-1;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	deep[1]=1;
	dfs1(1);
	dfs2(1,1);
	for(int i=1;i<=n;i++)
		insert(T[c[i]],1,n,id[i],w[i]);
	for(int i=1;i<=q;i++)
	{
		char s[3];
		cin>>s;
		if(s[0]=='C'&&s[1]=='C')
		{
			scanf("%d%d",&x,&y);
			delt(T[c[x]],1,n,id[x]);
			c[x]=y;
			insert(T[c[x]],1,n,id[x],w[x]);
		}
		if(s[0]=='C'&&s[1]=='W')
		{
			scanf("%d%d",&x,&y);
			w[x]=y;
			insert(T[c[x]],1,n,id[x],w[x]);
		}
		if(s[0]=='Q'&&s[1]=='S')
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",getsum(x,y));
		}
		if(s[0]=='Q'&&s[1]=='M')
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",getmax(x,y));
		}
	}
	return 0;
}