solution:
树链剖分,但是每个颜色要建一颗线段树
空间不够所以要动态开点
调了好久···
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cctype>
#define N 100005
using namespace std;
int n,q,cnt,head[N],w[N],c[N];
int tot,dep[N],fa[N],rk[N],siz[N],top[N],son[N],id[N];
inline int rd(){
int x=0,f=0;char ch=0;
while(!isdigit(ch)) f|=ch=='-',ch=getchar();
while( isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return f?-x:x;
}
struct EDGE{
int to,nxt;
}edge[N<<1];
inline void add(int x,int y){
edge[++cnt].nxt=head[x]; edge[cnt].to=y; head[x]=cnt;
}
inline void dfs1(int u,int fat,int d){
siz[u]=1; dep[u]=d; fa[u]=fat;
int maxson=-1;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to; if(v==fat) continue;
dfs1(v,u,d+1);
siz[u]+=siz[v];
if(siz[v]>maxson) son[u]=v,maxson=siz[v];
} return;
}
inline void dfs2(int u,int t){
top[u]=t; id[u]=++tot; rk[tot]=u;
if(!son[u]) return;
dfs2(son[u],t);
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to;
if(v!=son[u] && v!=fa[u]) dfs2(v,v);
} return;
}
struct TREE{
int ls,rs,sum,mx;
}node[N<<4];
int del[N],dlcnt,ndcnt,root[N];
inline void pushup(int cur){
int lson=node[cur].ls,rson=node[cur].rs;
node[cur].sum=node[lson].sum+node[rson].sum;
node[cur].mx=max(node[lson].mx,node[rson].mx);
}
inline int newnode(){
return dlcnt?del[dlcnt--]:++ndcnt;
}
inline void update(int &cur,int l,int r,int L,int R,int val){
if(!cur) cur=newnode();
if(l==r){
node[cur].sum=node[cur].mx=val;
return;
}
int mid=(l+r)>>1;
if(L<=mid) update(node[cur].ls,l,mid,L,R,val);
if(mid<R) update(node[cur].rs,mid+1,r,L,R,val);
pushup(cur);
if(node[cur].sum==0) del[++dlcnt]=cur,cur=0;
}
inline int query_sum(int cur,int l,int r,int L,int R){
if(!cur) return 0;
if(L<=l && r<=R) return node[cur].sum;
int mid=(l+r)>>1,res=0;
if(L<=mid) res+=query_sum(node[cur].ls,l,mid,L,R);
if(mid<R) res+=query_sum(node[cur].rs,mid+1,r,L,R);
return res;
}
inline int query_max(int cur,int l,int r,int L,int R){
if(!cur) return 0;
if(L<=l && r<=R) return node[cur].mx;
int mid=(l+r)>>1,res=0;
if(L<=mid) res=max(res,query_max(node[cur].ls,l,mid,L,R));
if(mid<R) res=max(res,query_max(node[cur].rs,mid+1,r,L,R));
return res;
}
inline int ask_sum(int x,int y){
int cl=c[x],ans=0,fx=top[x],fy=top[y];
while(fx!=fy){
if(dep[fx]>=dep[fy]){
ans+=query_sum(root[cl],1,n,id[fx],id[x]);
x=fa[fx];
}
else{
ans+=query_sum(root[cl],1,n,id[fy],id[y]);
y=fa[fy];
}
fx=top[x],fy=top[y];
}
if(id[x]<=id[y]) ans+=query_sum(root[cl],1,n,id[x],id[y]);
else ans+=query_sum(root[cl],1,n,id[y],id[x]);
return ans;
}
inline int ask_max(int x,int y){
int cl=c[x],ans=0,fx=top[x],fy=top[y];
while(fx!=fy){
if(dep[fx]>=dep[fy]){
ans=max(ans,query_max(root[cl],1,n,id[fx],id[x]));
x=fa[fx];
}
else{
ans=max(ans,query_max(root[cl],1,n,id[fy],id[y]));
y=fa[fy];
}
fx=top[x],fy=top[y];
}
if(id[x]<=id[y]) ans=max(ans,query_max(root[cl],1,n,id[x],id[y]));
else ans=max(ans,query_max(root[cl],1,n,id[y],id[x]));
return ans;
}
int main(){
n=rd(); q=rd();
for(int i=1;i<=n;i++) w[i]=rd(),c[i]=rd();
for(int i=1;i<n;i++){
int x=rd(),y=rd();
add(x,y); add(y,x);
}
dfs1(1,1,1); dfs2(1,1);
for(int i=1;i<=n;i++)
update(root[c[i]],1,n,id[i],id[i],w[i]);
while(q--){
char s[5];int x,y;
scanf("%s",s); x=rd(),y=rd();
if(s[0]=='C' && s[1]=='C') {
update(root[c[x]],1,n,id[x],id[x],0);
c[x]=y;
update(root[y],1,n,id[x],id[x],w[x]);
}
else if(s[0]=='C' && s[1]=='W') w[x]=y,update(root[c[x]],1,n,id[x],id[x],w[x]);
else if(s[0]=='Q' && s[1]=='S') printf("%d\n",ask_sum(x,y));
else printf("%d\n",ask_max(x,y));
return 0;
}
有个博客讲动态开点线段树和主席树什么的很好的