版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ccsu_cat/article/details/83063104
这个题要根据每个城市信仰的宗教建线段树,但是这样的话最多会有1e5个线段树,所以要动态开点(其实就是主席树思想),先把树用重链剖分,剖出来每个点有在线段树上对应的区间,所以线段树共用这一套区间,问题就简单了,至于修改城市的宗教,只需把原来宗教那个线段树的对应区间清0,在修改后宗教的那个线段树再加上一个新的点就行。
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=2e5+10;
int rt[maxn],ls[maxn*20],rs[maxn*20],Max[maxn*20],sum[maxn*20],id[maxn],rank[maxn];
int dep[maxn],son[maxn],f[maxn],top[maxn],w[maxn],c[maxn],size[maxn],n,cnt,tot;
vector<int>G[maxn];
void dfs(int u,int fa,int deep)
{
f[u]=fa;
size[u]=1;
dep[u]=deep;
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==fa)continue;
dfs(v,u,deep+1);
size[u]+=size[v];
if(size[v]>size[son[u]])son[u]=v;
}
}
void dfs2(int u,int root)
{
top[u]=root;
id[u]=++tot;
rank[tot]=u;
if(son[u])dfs2(son[u],root);
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 up(int &o,int l,int r,int k,int v)
{
if(!o)o=++cnt;
if(l==r)
{
Max[o]=sum[o]=v;
return;
}
int m=(l+r)/2;
if(k<=m)up(ls[o],l,m,k,v);
else up(rs[o],m+1,r,k,v);
Max[o]=max(Max[ls[o]],Max[rs[o]]);
sum[o]=sum[ls[o]]+sum[rs[o]];
}
int qsum(int o,int l,int r,int ql,int qr)
{
if(l>=ql&&r<=qr)return sum[o];
int m=(l+r)/2,res=0;
if(ql<=m)res+=qsum(ls[o],l,m,ql,qr);
if(qr>m)res+=qsum(rs[o],m+1,r,ql,qr);
return res;
}
int qmax(int o,int l,int r,int ql,int qr)
{
if(l>=ql&&r<=qr)return Max[o];
int m=(l+r)/2,res=0;
if(ql<=m)res=max(res,qmax(ls[o],l,m,ql,qr));
if(qr>m)res=max(res,qmax(rs[o],m+1,r,ql,qr));
return res;
}
int qu_path(int x,int y,int type)
{
int o=rt[c[x]];
int res[2]={0};
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
res[0]+=qsum(o,1,n,id[top[x]],id[x]);
res[1]=max(res[1],qmax(o,1,n,id[top[x]],id[x]));
x=f[top[x]];
}
if(id[x]>id[y])swap(x,y);
res[0]+=qsum(o,1,n,id[x],id[y]);
res[1]=max(res[1],qmax(o,1,n,id[x],id[y]));
return res[type];
}
int main()
{
int q,u,v;
char op[10];
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++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,0,1);
dfs2(1,1);
for(int i=1;i<=n;i++)
up(rt[c[i]],1,n,id[i],w[i]);
while(q--)
{
scanf("%s%d%d",op,&u,&v);
if(op[1]=='C')
{
up(rt[c[u]],1,n,id[u],0);
up(rt[v],1,n,id[u],w[u]);
c[u]=v;
}
else if(op[1]=='W')
w[u]=v,up(rt[c[u]],1,n,id[u],v);
else if(op[1]=='S')
printf("%d\n",qu_path(u,v,0));
else printf("%d\n",qu_path(u,v,1));
}
}