bzoj2243: [SDOI2011]染色

Description

给定一棵有n个节点的无根树和m个操作，操作有2类：

1、将节点a到节点b路径上所有点都染成颜色c；

2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），

如“112221”由3段组成：“11”、“222”和“1”。

请你写一个程序依次完成这m个操作。

Input

第一行包含2个整数n和m，分别表示节点数和操作数；

第二行包含n个正整数表示n个节点的初始颜色

下面 行每行包含两个整数x和y，表示x和y之间有一条无向边。

下面 行每行描述一个操作：

“C a b c”表示这是一个染色操作，把节点a到节点b路径上所有点（包括a和b）都染成颜色c；

“Q a b”表示这是一个询问操作，询问节点a到节点b（包括a和b）路径上的颜色段数量。

Output

对于每个询问操作，输出一行答案。

Sample Input

6 5
2 2 1 2 1 1
1 2
1 3
2 4
2 5
2 6
Q 3 5
C 2 1 1
Q 3 5
C 5 1 2
Q 3 5

Sample Output

3
1
2

HINT

数N<=10^5，操作数M<=10^5，所有的颜色C为整数且在[0, 10^9]之间。

题解：

树剖..维护区间左端点颜色，右端点颜色，然后维护区间颜色段数量，合并就判断一下左儿子的右端点右儿子的左端点颜色是否相同

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=300000+10;
struct node{
	int cnum,cl,cr;
	int lazy;
	node(){
		cnum=0;
		cl=cr=lazy=-1;
	}
}nd[maxn*8];
int top[maxn],size[maxn],son[maxn];
int fa[maxn],dep[maxn],tid[maxn];
int td=0;
vector<int>A[maxn];
int color[maxn];
int num[maxn];
int n;
inline void dfs1(int x,int d,int f){
	dep[x]=d;
	size[x]=1;
	fa[x]=f;
	for(int i=0;i<A[x].size();i++){
		int u=A[x][i];
		if(u==f)
			continue;
		dfs1(u,d+1,x);
		size[x]+=size[u];
		if(size[son[x]]<size[u])
			son[x]=u;
	}
}
inline void dfs2(int x,int tp){
	tid[x]=++td;
	top[x]=tp;
	if(!son[x])
		return ;
	dfs2(son[x],tp);
	for(int i=0;i<A[x].size();i++){
		int u=A[x][i];
		if(u==fa[x]||u==son[x])
			continue;
		dfs2(u,u);
	}
}
inline void pushdown(int o){
	if(nd[o].lazy!=-1){
		nd[o<<1].cnum=nd[o<<1|1].cnum=1;
		nd[o<<1].cl=nd[o<<1].cr=nd[o<<1|1].cl=nd[o<<1|1].cr=nd[o].lazy;
		nd[o<<1].lazy=nd[o<<1|1].lazy=nd[o].lazy;
		nd[o].lazy=-1;
	}
}
inline void pushup(int o){
	nd[o].cl=nd[o<<1].cl,nd[o].cr=nd[o<<1|1].cr;
	nd[o].cnum=nd[o<<1].cnum+nd[o<<1|1].cnum-(nd[o<<1].cr==nd[o<<1|1].cl);
}
inline void change(int o,int l,int r,int L,int R,int x){
	pushdown(o);
	if(L<=l&&r<=R){
		nd[o].lazy=x;
		nd[o].cl=nd[o].cr=x;
		nd[o].cnum=1;
		return ;
	}
	else {
		int mid=(l+r)>>1;
		if(mid>=L)
			change(o<<1,l,mid,L,R,x);
		if(mid<R)
			change(o<<1|1,mid+1,r,L,R,x);
		pushup(o);
	}
}
inline void build(int o,int l,int r){
	if(l==r){
		nd[o].cl=nd[o].cr=num[l];
		nd[o].lazy=-1,nd[o].cnum=1;
		return ;
	}
	else {
		int mid=(l+r)>>1;
		build(o<<1,l,mid);
		build(o<<1|1,mid+1,r);
		pushup(o);
	}
}
inline node query(int o,int l,int r,int L,int R){
	pushdown(o);
	if(L<=l&&r<=R)
		return nd[o];
	else {
		int mid=(l+r)>>1;
		node ll,rr,ans;
		if(mid>=L){
			ll=query(o<<1,l,mid,L,R);
			ans.cnum+=ll.cnum;
			ans.cl=ll.cl;
			ans.cr=ll.cr;
		}
		if(mid<R){
			rr=query(o<<1|1,mid+1,r,L,R);
			ans.cnum+=rr.cnum;
			if(ans.cl==-1)
				ans.cl=rr.cl;
			ans.cr=rr.cr;
		}
		if(ll.cnum&&rr.cnum){
			if(ll.cr==rr.cl)
				ans.cnum--;
		}
		return ans;
	}
}
inline int Query(int x,int y){
	int ans=0;
	int lstx=-1,lsty=-1;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]){
			swap(x,y);
			swap(lstx,lsty);
		}
		node ndd=query(1,1,n,tid[top[x]],tid[x]);
		ans+=ndd.cnum;
		if(lstx!=-1){
			if(ndd.cr==lstx)
				ans--;
		}
		lstx=ndd.cl;
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]){
		swap(x,y);
		swap(lstx,lsty);
	}
	node ndd=query(1,1,n,tid[x],tid[y]);
	ans+=ndd.cnum;
	if(lsty!=-1){
		if(ndd.cr==lsty)
			ans--;
	}
	if(lstx!=-1){
		if(ndd.cl==lstx)
			ans--;
	}
	return ans;
}
inline void Change(int x,int y,int z){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])
			swap(x,y);
		change(1,1,n,tid[top[x]],tid[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])
		swap(x,y);
	change(1,1,n,tid[x],tid[y],z);
}
int main(){
	//int size = 50 << 20;
	//char *p = (char*)malloc(size) + size;
	//__asm__("movl %0, %%esp\n" :: "r"(p));
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout);
	int m;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&color[i]);
	int x,y;
	for(int i=1;i<n;i++){
		scanf("%d %d",&x,&y);
		A[x].push_back(y);
		A[y].push_back(x);
	}
	dfs1(1,1,1);
	dfs2(1,1);
	for(int i=1;i<=n;i++)
		num[tid[i]]=color[i];
	build(1,1,n);
	char c;
	int z;
	while(m--){
		cin>>c;
		if(c=='C'){
			scanf("%d %d %d",&x,&y,&z);
			Change(x,y,z);
		}
		else {
			scanf("%d %d",&x,&y);
			printf("%d\n",Query(x,y));
		}
	}
return 0;
}

ps:我比较弱，bzoj上一直re,但luogu上ac了，如果有神犇知道是怎么回事请联系蒟蒻qwq