[SDOI2011]染色

这题主要是来告诫自己，除非写主席树和线段树合并，否则左右儿子都要访问（以下传数据）。

#include<cstdio>
#include<cstring>
#include<cctype>
#include<algorithm>
#define maxn 100005
#define lc now<<1
#define rc lc|1
using namespace std;

int n,m,ptcol[maxn];
int info[maxn],Prev[maxn*2],to[maxn*2],cnt_e;
inline void Node(int u,int v){ Prev[++cnt_e]=info[u],info[u]=cnt_e,to[cnt_e] =v;}

int siz[maxn],dep[maxn],id[maxn],tp[maxn],fa[maxn],son[maxn];
void dfs1(int now,int ff)
{
	dep[now]=dep[fa[now]=ff]+1;
	siz[now]=1;
	for(int i=info[now];i;i=Prev[i])
		if(to[i]!=ff)
		{
			dfs1(to[i],now);
			siz[now]+=siz[to[i]];
			(son[now]==-1 || siz[to[i]] > siz[son[now]]) && (son[now]=to[i]);
		}
}

int tim;
void dfs2(int now,int ff)
{
	id[now]=++tim;
	if(son[now]!=-1){ tp[son[now]]=tp[now];dfs2(son[now],now); }
	for(int i=info[now];i;i=Prev[i])
		if(to[i]!=ff && to[i]!=son[now])
		{
			tp[to[i]]=to[i];
			dfs2(to[i],now);
		}
}

int sum[maxn*10],lcol[maxn*10],rcol[maxn*10],col[maxn*10];//In fact, col is just a lazy tag!
inline void upd(int now)
{
	sum[now]=sum[lc]+sum[rc]-(rcol[lc] == lcol[rc]);
	lcol[now]=lcol[lc];rcol[now]=rcol[rc];
}
inline void dcd(int now)
{
	if(col[now]!=-1)
	{
		lcol[now]=rcol[now]=lcol[lc]=rcol[lc]=lcol[rc]=rcol[rc]=col[lc]=col[rc]=col[now];
		sum[now]=1;
		col[now]=-1;
	}
}

void Insert(int now,int l,int r,int ql,int qr,int c)
{
	dcd(now);
	if(qr<l || ql>r) return;
	if(ql<= l && r<=qr)
	{
		col[now]=c;
		dcd(now);
		return;
	}
	int mid=(l+r)>>1;
	Insert(lc,l,mid,ql,qr,c);
	Insert(rc,mid+1,r,ql,qr,c);
	upd(now);
}

void Insert(int u,int v,int c)
{
	for(;tp[u]!=tp[v];)
	{
		if(dep[tp[u]]>dep[tp[v]]) swap(u,v);
		Insert(1,1,n,id[tp[v]],id[v],c);
		v=fa[tp[v]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	Insert(1,1,n,id[u],id[v],c);
}

int Query(int now,int l,int r,int ql,int qr)
{
	dcd(now);
	if(l>qr || r<ql) return 0;
	if(ql<=l && r<=qr) return sum[now];
	int mid=(l+r)>>1;
	int r1=Query(lc,l,mid,ql,qr),r2=Query(rc,mid+1,r,ql,qr);
	upd(now);
	return r1 + r2 - (r1 && r2 && rcol[lc]==lcol[rc]);
}

int Findc(int now,int l,int r,int pos)
{
	dcd(now);
	if(pos<l || pos>r) return 0;
	if(l==r) return lcol[now];
	int mid=(l+r)>>1,tmp;
	tmp=Findc(lc,l,mid,pos) + Findc(rc,mid+1,r,pos);
	upd(now);
	return tmp;
}

int Query(int u,int v)
{
	int ret=0;
	for(;tp[u]!=tp[v];)
	{
		if(dep[tp[u]]>dep[tp[v]]) swap(u,v);
		ret+=Query(1,1,n,id[tp[v]],id[v])-(Findc(1,1,n,id[tp[v]])==Findc(1,1,n,id[fa[tp[v]]]));
		v=fa[tp[v]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	ret+=Query(1,1,n,id[u],id[v]);
	return ret;
}

int main()
{
	/*
	#ifndef ONLINE_JUDGE
	freopen("2.in","r",stdin);
	#endif
	*/
	int u,v,w;
	memset(son,-1,sizeof son);
	memset(col,-1,sizeof col);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&ptcol[i]);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		Node(u,v),Node(v,u);
	}
	
	dfs1(1,0);tp[1]=1;
	dfs2(1,0);
	
	for(int i=1;i<=n;i++) Insert(1,1,n,id[i],id[i],ptcol[i]);
		
	char ch[5];
	for(;m--;)
	{
		scanf("%s",ch);
		if(ch[0]=='C')
		{
			scanf("%d%d%d",&u,&v,&w);
			Insert(u,v,w);
		}
		else 
		{
			scanf("%d%d",&u,&v);
			printf("%d\n",Query(u,v));
		}
	}
}