这道题学到了动态开空间线段树的写法,感觉跟主席树有一定的类似。
说一下怎么动态开吧。
首先回忆一下线段树的过程,我们发现对于这题,每一颗线段树都有很多的子结点是0,也就是说这些节点都是没用的,所以我们考虑对于每一个非0叶子结点到根都是非0的,但是其他都是为0的,都是没有用的,那么为何要为它们开内存存储呢,所以我们换种方式存储对每一个节点都有一个编号,然后结构体内的l,r都是存的l,r的编号,对于没用的节点都不开内存给它。
再说下这题的题解,开c个线段树来维护c个宗教,所以每一宗教就直接树链剖分,然后动态开内存空间要开大点。
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const int Maxn=1000005;
struct node{int v,next;}Qt[Maxn*2];
struct segment{int l,r,mx,sum;}t[Maxn*4];
int n,m,cnt,tmp,SIZE;
int w[Maxn],c[Maxn],head[Maxn],size[Maxn],fa[Maxn],depth[Maxn],pos[Maxn],top[Maxn],root[Maxn];
inline void swap(int &a,int &b){int t=a;a=b;b=t;}
inline int max(int a,int b){return a>b?a:b;}
inline void Insert(int u,int v)
{
cnt++;
Qt[cnt].v=v;
Qt[cnt].next=head[u];
head[u]=cnt;
}
void dfs1(int x)
{
size[x]=1;
for(int i=head[x];i;i=Qt[i].next)
{
if(Qt[i].v==fa[x])continue;
depth[Qt[i].v]=depth[x]+1;
fa[Qt[i].v]=x;
dfs1(Qt[i].v);
size[x]+=size[Qt[i].v];
}
}
void dfs2(int x,int tp)
{
int k=0;
pos[x]=++tmp;
top[x]=tp;
for(int i=head[x];i;i=Qt[i].next)
if(depth[Qt[i].v]>depth[x]&&size[Qt[i].v]>size[k])
k=Qt[i].v;
if(k)
dfs2(k,tp);
for(int i=head[x];i;i=Qt[i].next)
if(depth[Qt[i].v]>depth[x]&&Qt[i].v!=k)
dfs2(Qt[i].v,Qt[i].v);
}
inline void change(int &k,int l,int r,int x,int d)
{
if(!k)k=++SIZE;
if(l==r){t[k].mx=t[k].sum=d;return ;}
int mid=(l+r)>>1;
if(x<=mid)change(t[k].l,l,mid,x,d);
else change(t[k].r,mid+1,r,x,d);
t[k].mx=max(t[t[k].l].mx,t[t[k].r].mx);
t[k].sum=t[t[k].l].sum+t[t[k].r].sum;
}
inline int Qmax(int k,int l,int r,int lf,int rf)
{
if(l>=lf&&r<=rf)return t[k].mx;
int mid=(l+r)>>1,mx=0;
if(mid>=lf)mx=max(mx,Qmax(t[k].l,l,mid,lf,rf));
if(mid<rf)mx=max(mx,Qmax(t[k].r,mid+1,r,lf,rf));
return mx;
}
inline int Qsum(int k,int l,int r,int lf,int rf)
{
if(l>=lf&&r<=rf)return t[k].sum;
int mid=(l+r)>>1,sum=0;
if(mid>=lf)sum+=Qsum(t[k].l,l,mid,lf,rf);
if(mid<rf)sum+=Qsum(t[k].r,mid+1,r,lf,rf);
return sum;
}
inline int solvesum(int _c,int x,int y)
{
int sum=0;
while(top[x]!=top[y])
{
if(depth[top[x]]<depth[top[y]])swap(x,y);
sum+=Qsum(root[_c],1,n,pos[top[x]],pos[x]);
x=fa[top[x]];
}
if(pos[x]>pos[y])swap(x,y);
sum+=Qsum(root[_c],1,n,pos[x],pos[y]);
return sum;
}
inline int solvemax(int _c,int x,int y)
{
int mx=0;
while(top[x]!=top[y])
{
if(depth[top[x]]<depth[top[y]])swap(x,y);
mx=max(mx,Qmax(root[_c],1,n,pos[top[x]],pos[x]));
x=fa[top[x]];
}
if(pos[x]>pos[y])swap(x,y);
mx=max(mx,Qmax(root[_c],1,n,pos[x],pos[y]));
return mx;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d%d",&w[i],&c[i]);
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
Insert(x,y);
Insert(y,x);
}
dfs1(1);
dfs2(1,1);
for(int i=1;i<=n;i++)
change(root[c[i]],1,n,pos[i],w[i]);
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
char ch[5];
int x,y;
scanf("%s",ch);
scanf("%d%d",&x,&y);
if(ch[1]=='C')
change(root[c[x]],1,n,pos[x],0),c[x]=y,change(root[c[x]],1,n,pos[x],w[x]);
else if(ch[1]=='W')
change(root[c[x]],1,n,pos[x],y),w[x]=y;
else if(ch[1]=='S')
printf("%d\n",solvesum(c[x],x,y));
else if(ch[1]=='M')
printf("%d\n",solvemax(c[x],x,y));
}
return 0;
}