题解 [SDOI2014] 旅行（LOJ #2195 / 洛谷 P3313）【树链剖分 动态开点线段树】

题目链接：洛谷 P3313 / LOJ #2195

题意

一棵 $n$ 个点的数，每个点有一个颜色 $c_i$ 与权值 $w_i$，$m$ 次操作：

• 修改某个点的权值；
• 修改某个点的颜色；
• 查询 $x$ 到 $y$ 的路径上，颜色为 $c_x$ 的所有点的权值和（保证 $c_x=c_y$）；
• 查询 $x$ 到 $y$ 的路径上，颜色为 $c_x$ 的所有点的最大值（保证 $c_x=c_y$）。

$n,m,c\le 10^5$。

题解

最近一直在写水题来回忆起远古时期看过的算法。

不过好像是我第一次写动态开点的线段树。以前大部分情况都写 01 Trie 糊过去的。

假如没有颜色限制，这就是树剖裸题。

现在有颜色限制，就给每个颜色建一棵线段树。

直接建空间会炸，故动态开点。

代码：

/**********
Author: WLBKR5
Problem: loj 2195, luogu 3313
Name: 旅行 
Source: SDOI2014 
Algorithm: 树链剖分, 动态开点线段树 
Date: 2020/06/15
Statue: accepted
Submission: loj.ac/submission/836580, www.luogu.com.cn/record/34411338
**********/
#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const int N=1e5+10;
struct bian{
	int e,n;
};
bian b[N<<1];
int s[N],tot=0;
void add(int x,int y){
	++tot;
	b[tot].e=y;
	b[tot].n=s[x];
	s[x]=tot;
}
int col[N],val[N];
int n;

int rt[N],ch[N<<5][2],sum[N<<5],mx[N<<5],cnt=0;
void pushup(int x){
	sum[x]=sum[ch[x][0]]+sum[ch[x][1]];
	mx[x]=max(mx[ch[x][0]],mx[ch[x][1]]);
}
void modify(int pos,int v,int &x,int nl,int nr){
	if(!x)x=++cnt;
	if(nl==nr){
		mx[x]=v;
		sum[x]=v;
		return;
	}
	int mid=nl+nr>>1;
	if(pos<=mid)modify(pos,v,ch[x][0],nl,mid);
	else modify(pos,v,ch[x][1],mid+1,nr);
	pushup(x);
}
int querys(int l,int r,int x,int nl,int nr){
	if(!x)return 0;
	if(nr<l||nl>r)return 0;
	if(l<=nl&&nr<=r)return sum[x];
	int mid=nl+nr>>1;
	return querys(l,r,ch[x][0],nl,mid)+querys(l,r,ch[x][1],mid+1,nr);
}
int querym(int l,int r,int x,int nl,int nr){
	if(!x)return 0;
	if(nr<l||nl>r)return 0;
	if(l<=nl&&nr<=r)return mx[x];
	int mid=nl+nr>>1;
	return max(querym(l,r,ch[x][0],nl,mid),querym(l,r,ch[x][1],mid+1,nr));
}

int dep[N],sz[N],maxson[N],fa[N];
int dfn[N],dfnend[N],dfnn=0;
int top[N];
void ss1(int x,int f){
	sz[x]=1;
	for(int i=s[x];i;i=b[i].n){
		if(b[i].e==f)continue;
		dep[b[i].e]=dep[x]+1;
		fa[b[i].e]=x;
		ss1(b[i].e,x);
		sz[x]+=sz[b[i].e];
		if(sz[b[i].e]>sz[maxson[x]])maxson[x]=b[i].e;
	}
}
void ss2(int x,int f,int t){
	top[x]=t;
	dfn[x]=dfnend[x]=++dfnn;
	if(maxson[x])ss2(maxson[x],x,t);
	for(int i=s[x];i;i=b[i].n){
		if(b[i].e==f||b[i].e==maxson[x])continue;
		ss2(b[i].e,x,b[i].e);
		dfnend[x]=dfnend[b[i].e];
	}
}
int qs(int x,int y){
	int c=col[x];
	int sum=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		sum+=querys(dfn[top[x]],dfn[x],rt[c],1,n);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])swap(x,y);
	sum+=querys(dfn[y],dfn[x],rt[c],1,n);
	return sum;
}
int qm(int x,int y){
	int c=col[x];
	int mx=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		mx=max(mx,querym(dfn[top[x]],dfn[x],rt[c],1,n));
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])swap(x,y);
	mx=max(mx,querym(dfn[y],dfn[x],rt[c],1,n));
	return mx;
}

char tmp[5];
int main(){
	n=getint();
	int m=getint();
	for(int i=1;i<=n;i++){
		val[i]=getint();
		col[i]=getint();
	}
	for(int i=1;i<n;i++){
		int x=getint(),y=getint();
		add(x,y);
		add(y,x);
	}
	dep[1]=1;
	ss1(1,1);
	ss2(1,0,1);
	for(int i=1;i<=n;i++){
		modify(dfn[i],val[i],rt[col[i]],1,n);
	}
	for(int i=0;i<m;i++){
		scanf("%s",tmp);
		if(tmp[0]=='C'&&tmp[1]=='C'){
			int x=getint(),c=getint();
			modify(dfn[x],0,rt[col[x]],1,n);
			col[x]=c;
			modify(dfn[x],val[x],rt[col[x]],1,n);
		}
		if(tmp[0]=='C'&&tmp[1]=='W'){
			int x=getint(),w=getint();
			val[x]=w;
			modify(dfn[x],val[x],rt[col[x]],1,n);
		}
		if(tmp[0]=='Q'&&tmp[1]=='S'){
			int x=getint(),y=getint();
			printf("%d\n",qs(x,y));
		}
		if(tmp[0]=='Q'&&tmp[1]=='M'){
			int x=getint(),y=getint();
			printf("%d\n",qm(x,y));
		}
	}
	return 0;
}