洛谷P3313 [SDOI2014]旅行（树剖+线段树动态开点）

洛谷P3313 [SDOI2014]旅行（树剖+线段树动态开点）

题意：

“CC x c“：城市x的居民全体改信了c教；

“CW x w“：城市x的评级调整为w;

“QS x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级总和；

“QM x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级最大值。

思路：

对每种信教开一颗线段树，动态开点

#include <bits/stdc++.h>
using namespace std;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
typedef long long ll;
const int maxn=1e5+10;
int n,q,r,mod,len,tot,a[maxn],be[maxn],siz[maxn],fa[maxn],son[maxn];
int id[maxn],idd[maxn],top[maxn],deep[maxn],root[maxn];
struct node{
	int l,r,mx,sum;
}tr[maxn*400];
vector<int>mp[maxn];
void dfs1(int x,int f,int d){
	siz[x]=1;
	fa[x]=f;
	deep[x]=d;
	son[x]=0;
	for(auto y:mp[x]){
		if(y==f)continue;
		dfs1(y,x,d+1);
		siz[x]+=siz[y];
		if(siz[son[x]]<siz[y])son[x]=y;
	}
}
void dfs2(int x,int root){
	id[x]=++tot;
	idd[tot]=x;
	top[x]=root;
	if(son[x])dfs2(son[x],root);
	for(auto y:mp[x]){
		if(y!=fa[x]&&y!=son[x])dfs2(y,y);
	}
}
void pushup(int now){
	tr[now].sum=tr[tr[now].l].sum+tr[tr[now].r].sum;
	tr[now].mx=max(tr[tr[now].l].mx,tr[tr[now].r].mx);
}
/*
int update(int pre,int l,int r,int x,int val){
	int now=++cnt;
	tr[now]=tr[pre];
	if(l==r){
		tr[now].mx=tr[now].sum=val;
		return now;
	}
	int mid=(l+r)>>1;
	if(x<=mid)tr[now].l=update(tr[pre].l,l,mid,x,val);
	else tr[now].r=(tr[pre].r,mid+1,r,x,val);
	pushup(now);
	return now;
}
*/ 
inline void update(int &rt,int l,int r,int pos,int w){
    if (!rt) rt=++len;
    if (l==r){
    	tr[rt].mx=tr[rt].sum=w;
    	return;
	} 
	int mid=(l+r)/2;
    if (mid>=pos) update(tr[rt].l,l,mid,pos,w);
    else update(tr[rt].r,mid+1,r,pos,w);
    pushup(rt);
    return;
}

int sum_query(int now,int l,int r,int x,int y){
	if(x<=l&&r<=y)return tr[now].sum;
	int mid=(l+r)>>1,ans=0;
	if(x<=mid)ans+=sum_query(tr[now].l,l,mid,x,y);
	if(y>mid)ans+=sum_query(tr[now].r,mid+1,r,x,y);
	return ans;
}
int max_query(int now,int l,int r,int x,int y){
	if(x<=l&&r<=y)return tr[now].mx;
	int mid=(l+r)>>1,ans=0;
	if(x<=mid)ans=max(ans,max_query(tr[now].l,l,mid,x,y));
	if(y>mid)ans=max(ans,max_query(tr[now].r,mid+1,r,x,y));
	return ans;
}
int getdis(int x,int y,int kind){//点1到x的权值和 
	int ans=0;
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]])swap(x,y);
		ans+=sum_query(root[kind],1,n,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(deep[x]>deep[y])swap(x,y);
	ans+=sum_query(root[kind],1,n,id[x],id[y]);
	return ans;
}
int getmax(int x,int y,int kind){//点1到x的权值和 
	int ans=0;
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]])swap(x,y);
		ans=max(ans,max_query(root[kind],1,n,id[top[x]],id[x]));
		x=fa[top[x]];
	}
	if(deep[x]>deep[y])swap(x,y);
	ans=max(ans,max_query(root[kind],1,n,id[x],id[y]));
	return ans;
}
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)
	    scanf("%d%d",&a[i],&be[i]);
	for(int i=1,x,y;i<n;i++){
		scanf("%d%d",&x,&y);
		mp[x].push_back(y),mp[y].push_back(x);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	for(int i=1;i<=n;i++)
	    update(root[be[i]],1,n,id[i],a[i]);
	while(q--){
	    char op[4];
		int x,y;
		scanf("%s%d%d",op,&x,&y);
		if(op[1]=='C'){
			update(root[be[x]],1,n,id[x],0);
			be[x]=y;
			update(root[be[x]],1,n,id[x],a[x]);
		}
		else if(op[1]=='W'){
			update(root[be[x]],1,n,id[x],0);
			a[x]=y;
			update(root[be[x]],1,n,id[x],a[x]);
		}
		else if(op[1]=='S'){
			printf("%d\n",getdis(x,y,be[x]));
		}
		else if(op[1]=='M'){
			printf("%d\n",getmax(x,y,be[x]));
		}
	}
}