4999: This Problem Is Too Simple！

Description

给您一颗树，每个节点有个初始值。

现在支持以下两种操作：

1. C i x(0<=x<2^31) 表示将i节点的值改为x。

2. Q i j x(0<=x<2^31) 表示询问i节点到j节点的路径上有多少个值为x的节点。

题解：

先树剖，再对于每个值，开棵线段树，

一般这种题，都是这个套路吧，

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
const int N=101000;
struct node{
	int x,y,next;
}sa[N*2];int len=0,first[N];
int n,m;
int a[N];
void ins(int x,int y)
{
	len++;
	sa[len].x=x;
	sa[len].y=y;
	sa[len].next=first[x];
	first[x]=len;
}
int dep[N],fa[N],siz[N],son[N];
void dfs1(int x)
{
	siz[x]=1;
	for(int i=first[x];i!=-1;i=sa[i].next)
	{
		int y=sa[i].y;
		if(y!=fa[x]) 
		{
			dep[y]=dep[x]+1;
			fa[y]=x;
			dfs1(y);
			siz[x]+=siz[y];
			if(siz[son[x]]<siz[y]) son[x]=y;
		}
	}
}
int top[N],pos[N],yz=0;
void dfs2(int x,int tp)
{
	pos[x]=++yz;
	top[x]=tp;
	if(son[x]!=0) dfs2(son[x],tp);
	for(int i=first[x];i!=-1;i=sa[i].next)
	{
		int y=sa[i].y;
		if(y!=fa[x]&&y!=son[x])
		dfs2(y,y);
	}
}
int root[N<<2],ls[N<<7],rs[N<<7],sum[N<<7],clo=0,tot=0;
map<int,int>mp;
void update(int rt,int a,int l,int r,int &x)
{
	if(!x) x=++tot;
	sum[x]+=a;
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(rt<=mid) update(rt,a,l,mid,ls[x]);
	else update(rt,a,mid+1,r,rs[x]);
}
int query(int x,int y,int l,int r,int z)
{
	//printf("!");
	if(!z) return 0;
	//printf("!");
	if(x<=l&&r<=y) return sum[z];
	int mid=(l+r)>>1,ans=0;
	if(x<=mid) ans+=query(x,y,l,mid,ls[z]);
	if(y>mid)  ans+=query(x,y,mid+1,r,rs[z]);
	return ans;
}
int slove(int x,int y,int z)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ans+=query(pos[top[x]],pos[x],1,n,root[z]),x=fa[top[x]];
		//printf("!");
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans+=query(pos[x],pos[y],1,n,root[z]);
	return ans;
}
int main()
{
	scanf("%d%d",&n,&m);
	int x,y;
	memset(first,-1,sizeof(first));
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n-1;i++)
	{
		scanf("%d%d",&x,&y);
		ins(x,y);ins(y,x);
	}
	dfs1(1);dfs2(1,1);
	for(int i=1;i<=n;i++)
	{
		if(!mp[a[i]]) mp[a[i]]=++clo;
		update(pos[i],1,1,n,root[mp[a[i]]]);
	}
	char s[10];
	int z;
	for(int i=1;i<=m;i++)
	{
		scanf("%s%d%d",s,&x,&y);
		if(s[0]=='C')
		{
			update(pos[x],-1,1,n,root[mp[a[x]]]);
			a[x]=y;if(!mp[y]) mp[y]=++clo;
			update(pos[x],1,1,n,root[mp[y]]);
		}
		else
		{
			scanf("%d",&z);
			printf("%d\n",slove(x,y,mp[z]));
		}
	}
}