bzoj 4999: This Problem Is Too Simple！

题意：

给您一颗树，每个节点有个初始值。
现在支持以下两种操作：
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));        }    }}