题意:
给您一颗树,每个节点有个初始值。
现在支持以下两种操作:
1. C i x(0<=x<2^31) 表示将i节点的值改为x。
2. Q i j x(0<=x<2^31) 表示询问i节点到j节点的路径上有多少个值为x的节点。
题解:
看到这种东西,首先想到的就是树剖了吧。
但假如对于每一种值都开一棵线段树,空间爆炸。
但动态开点就没问题了。
关于值比较大,用map就行了。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
struct TRnode{
int fa,son,tot,top,dep;
}TR[100010];int ys[100010],z=0;
struct trnode{
int lc,rc,c;
}tr[5000010];int tot=0,root[300010];
map <int,int>s;int num=0;
int color[100010],n,m;
struct node{
int y,next;
}a[200010];int len=0,last[100010];
void ins(int x,int y)
{
a[++len].y=y;
a[len].next=last[x];last[x]=len;
}
void pre_node(int x,int fa)
{
TR[x].fa=fa;TR[x].dep=TR[fa].dep+1;TR[x].tot=1;
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==fa) continue;
pre_node(y,x);
if(TR[y].tot>TR[TR[x].son].tot) TR[x].son=y;
TR[x].tot+=TR[y].tot;
}
}
void pre_edge(int x,int top)
{
ys[x]=++z;TR[x].top=top;
if(TR[x].son) pre_edge(TR[x].son,top);
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==TR[x].fa||y==TR[x].son) continue;
pre_edge(y,y);
}
}
void update(int &x,int l,int r,int k,int c)
{
if(!x) x=++tot;
if(l==r){tr[x].c+=c;return;}
int mid=(l+r)/2;
if(k<=mid) update(tr[x].lc,l,mid,k,c);
else update(tr[x].rc,mid+1,r,k,c);
tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
}
int findans(int &x,int l,int r,int fl,int fr)
{
if(!x) return 0;
if(l==fl&&r==fr) return tr[x].c;
int mid=(l+r)/2;
if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
return findans(tr[x].lc,l,mid,fl,mid)+findans(tr[x].rc,mid+1,r,mid+1,fr);
}
int solve(int froot,int x,int y)
{
int tx=TR[x].top,ty=TR[y].top,ans=0;
while(tx!=ty)
{
if(TR[tx].dep<TR[ty].dep) swap(x,y),swap(tx,ty);
ans+=findans(froot,1,n,ys[tx],ys[x]);
x=TR[tx].fa;tx=TR[x].top;
}
if(TR[x].dep<TR[y].dep) swap(x,y);
ans+=findans(froot,1,n,ys[y],ys[x]);
return ans;
}
int main()
{
s.clear();
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
{
int x;scanf("%d",&x);
if(!s[x]) s[x]=++num;
color[i]=s[x];
}
for(int i=1;i<n;i++)
{
int x,y;scanf("%d %d",&x,&y);
ins(x,y);ins(y,x);
}
pre_node(1,0);pre_edge(1,1);
for(int i=1;i<=n;i++) update(root[color[i]],1,n,ys[i],1);
while(m--)
{
char S[3];scanf("%s",S);
if(S[0]=='C')
{
int i,x;scanf("%d %d",&i,&x);
if(!s[x]) s[x]=++num;
update(root[color[i]],1,n,ys[i],-1);
color[i]=s[x];
update(root[color[i]],1,n,ys[i],1);
}
else
{
int i,j,x;scanf("%d %d %d",&i,&j,&x);
if(!s[x]) s[x]=++num;
printf("%d\n",solve(root[s[x]],i,j));
}
}
}