【bzoj3531】[Sdoi2014]旅行 动态开点的线段树

动态开点的线段树
开c棵线段树
CC 对应删除一个节点，增加一个节点
CW 对应修改一个节点
QS 查询对应线段树中部分的和

QM 查询对应线段树中部分的最大值

所谓动态开点，就是每个节点用的时候再开，可以去掉许多无用的节点。

总共O(nlogn)级别个节点

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 200010
#define N 10000100 

using namespace std;

struct yts
{
	int lch,rch,mx,sum,l,r;
}t[N];

int root[maxn],head[maxn],to[maxn],next[maxn],w[maxn],c[maxn],st[maxn];
int fa[20][maxn],dep[maxn],seq[maxn],size[maxn],dd[maxn],e[maxn],rank[maxn];
bool vis[maxn];
char s[5];
int n,m,num,tot,T;

void addedge(int x,int y)
{
	num++;to[num]=y;next[num]=st[x];st[x]=num;
}

void dfs(int x)
{
	e[++tot]=x;size[x]=1;
	for (int p=st[x];p;p=next[p])
	  if (to[p]!=fa[0][x])
	  {
	  	fa[0][to[p]]=x;
	  	dep[to[p]]=dep[x]+1;
	  	dfs(to[p]);
	  	size[x]+=size[to[p]];
	  	if (size[to[p]]>size[dd[x]]) dd[x]=to[p];
	  }
}

int go_up(int x,int d)
{
	for (int i=17;i>=0;i--)
	  if (d&(1<<i)) x=fa[i][x];
	return x;
}

int LCA(int x,int y)
{
	if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
	else y=go_up(y,dep[y]-dep[x]);
	if (x==y) return x;
	for (int i=17;i>=0;i--)
	  if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
	return fa[0][x];
}

void update(int i)
{
	t[i].mx=max(t[t[i].lch].mx,t[t[i].rch].mx);
	t[i].sum=t[t[i].lch].sum+t[t[i].rch].sum;
}

void modify(int &i,int l,int r,int x,int d)
{
	if (!i)
	{
		i=++tot;
		t[i].lch=t[i].rch=t[i].mx=t[i].sum=0;
		t[i].l=l;t[i].r=r;
	}
	if (l==r) 
	{
		t[i].mx=t[i].sum=d;
		return;
	}
	int mid=(t[i].l+t[i].r)/2;
	if (x<=mid) modify(t[i].lch,l,mid,x,d);
	if (mid<x) modify(t[i].rch,mid+1,r,x,d);
	update(i);
}

int query_mx(int &i,int l,int r)
{
	if (!i) return 0;
	if (l<=t[i].l && t[i].r<=r) return t[i].mx;
	int mid=(t[i].l+t[i].r)/2,ans=0;
	if (l<=mid) ans=max(ans,query_mx(t[i].lch,l,r));
	if (mid<r) ans=max(ans,query_mx(t[i].rch,l,r));
	return ans;
}

int query_sum(int &i,int l,int r)
{
	if (!i) return 0;
	if (l<=t[i].l && t[i].r<=r) return t[i].sum;
	int mid=(t[i].l+t[i].r)/2,ans=0;
	if (l<=mid) ans+=query_sum(t[i].lch,l,r);
	if (mid<r) ans+=query_sum(t[i].rch,l,r);
	return ans;
}

int main()
{
	scanf("%d%d",&n,&T);
	for (int i=1;i<=n;i++) scanf("%d%d",&w[i],&c[i]);
	for (int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		addedge(x,y);addedge(y,x);
	}
	dfs(1);
	int qwer=0;
	for (int i=1;i<=n;i++)
	  if (!vis[e[i]])
	  {
	  	int k=e[i];
	  	while (k)
	  	{
	  		seq[++qwer]=k;head[k]=e[i];vis[k]=1;k=dd[k];
	  	}
	  }
	tot=0;
	for (int i=1;i<=n;i++) rank[seq[i]]=i;
	for (int j=1;j<=17;j++)
	  for (int i=1;i<=n;i++)
	    fa[j][i]=fa[j-1][fa[j-1][i]];
	for (int i=1;i<=n;i++) modify(root[c[i]],1,n,rank[i],w[i]);
	while (T--)
	{
		int x,y;
		scanf("%s%d%d",s,&x,&y);
		if (s[1]=='C')
		{
			modify(root[c[x]],1,n,rank[x],0);
			c[x]=y;
			modify(root[c[x]],1,n,rank[x],w[x]);
		}
		if (s[1]=='W')
		{
			w[x]=y;
			modify(root[c[x]],1,n,rank[x],w[x]);
		}
		if (s[1]=='M')
		{
			int z=LCA(x,y),col=c[x],ans=0;
			while (dep[head[x]]>dep[z])
			{
				ans=max(ans,query_mx(root[col],rank[head[x]],rank[x]));x=fa[0][head[x]];
			}
			ans=max(ans,query_mx(root[col],rank[z],rank[x]));
			while (dep[head[y]]>dep[z])
			{
				ans=max(ans,query_mx(root[col],rank[head[y]],rank[y]));y=fa[0][head[y]];
			}
			ans=max(ans,query_mx(root[col],rank[z],rank[y]));
			printf("%d\n",ans);
		}
		if (s[1]=='S')
		{
			int z=LCA(x,y),col=c[x],ans=0;
			while (dep[head[x]]>dep[z])
			{
				ans+=query_sum(root[col],rank[head[x]],rank[x]);x=fa[0][head[x]];
			}
			ans+=query_sum(root[col],rank[z],rank[x]);
			while (dep[head[y]]>dep[z])
			{
				ans+=query_sum(root[col],rank[head[y]],rank[y]);y=fa[0][head[y]];
			}
			ans+=query_sum(root[col],rank[z],rank[y]);
			ans-=query_sum(root[col],rank[z],rank[z]);
			printf("%d\n",ans);
		}
	}
	return 0;
}