题目描述
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;
}