树链剖分+动态开点线段树--luoguP3313 [SDOI2014]旅行

传送门

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;
}

有个博客讲动态开点线段树和主席树什么的很好的

https://www.cnblogs.com/keshuqi/p/6251525.html