[SDOI2011]染色

题目

题目描述
给定一棵 nn 个节点的无根树，共有 mm 个操作，操作分为两种：

将节点 aa 到节点 bb 的路径上的所有点（包括 aa 和 bb）都染成颜色 cc。
询问节点 aa 到节点 bb 的路径上的颜色段数量。
颜色段的定义是极长的连续相同颜色被认为是一段。例如 112221 由三段组成：11、222、1。

输入格式
输入的第一行是用空格隔开的两个整数，分别代表树的节点个数 nn 和操作个数 mm。

第二行有 nn 个用空格隔开的整数，第 ii 个整数 w_iw
i
​
代表结点 ii 的初始颜色。

第 33 到第 (n + 1)(n+1) 行，每行两个用空格隔开的整数 u, vu,v，代表树上存在一条连结节点 uu 和节点 vv 的边。

第 (n + 2)(n+2) 到第 (n + m + 1)(n+m+1) 行，每行描述一个操作，其格式为：

每行首先有一个字符 opop，代表本次操作的类型。

若 opop 为 C，则代表本次操作是一次染色操作，在一个空格后有三个用空格隔开的整数 a, b, ca,b,c，代表将 aa 到 bb 的路径上所有点都染成颜色 cc。

若 opop 为 Q，则代表本次操作是一次查询操作，在一个空格后有两个用空格隔开的整数 a, ba,b，表示查询 aa 到 bb 路径上的颜色段数量。

输出格式
对于每次查询操作，输出一行一个整数代表答案。

输入输出样例
输入 #1 复制
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
输出 #1 复制
3
1
2
说明/提示
数据规模与约定
对于 100%100% 的数据，1 \leq n, m \leq 10^51≤n,m≤10
5
，1 \leq w_i, c \leq 10^91≤w
i
​
,c≤10
9
，1 \leq a, b, u, v \leq n1≤a,b,u,v≤n，opop 一定为 C 或 Q，保证给出的图是一棵树。

除原数据外，还存在一组不计分的 hack 数据。

思路

树剖，用线段树维护：

数组tot[N]表示此时的颜色段数。

数组zzz[N]表示此时最左边的节点的颜色。

数组yyy[N]表示此时最右边的节点的颜色。

则： 首先我们要明确，线段树的叶子节点一定只有一种颜色，也就是一条颜色段。

tot[父亲]=tot[左儿子]+tot[右儿子]；

if (zzz[右儿子]==yyy[左儿子]) {tot[父亲]–;}

即如果右儿子的最左边颜色和左儿子的最右边颜色相同，那么肯定有中间部分属于同一颜色段。

代码

#include<bits.stdc++.h>
#define ll long long 
using namespace std;   
const int sigma_size=26;  
const int N=100+50;  
const int MAXN=100000+50;  
const int inf=0x3fffffff;  
const double eps=1e-8;  
const int mod=100000000+7;  
#define L(x) (x<<1)  
#define R(x) (x<<1|1)  
#define PII pair<int, int>  
#define mk(x,y) make_pair((x),(y))  
int n,m,edge_cnt,cnt,Lc,Rc;  
char str[N];  
int col[MAXN],head[MAXN],sz[MAXN],dep[MAXN],fa[MAXN],son[MAXN],top[MAXN],pos[MAXN];  
struct Edge{  
	int u,v,next;  
}edge[MAXN<<1];  
struct node{  
	int l,r;  
	int num,tag,lc,rc;  
}segtree[MAXN<<2];  
void init(){  
	edge_cnt=cnt=0;  
	memset(head,-1,sizeof(head));  
}  
void addedge(int u,int v){  
	edge[edge_cnt].u=u; edge[edge_cnt].v=v;  
	edge[edge_cnt].next=head[u]; head[u]=edge_cnt++;  
}  
void dfs1(int u,int pre,int depth){  
	sz[u]=1; fa[u]=pre; son[u]=0; dep[u]=depth;  
	for(int i=head[u];i!=-1;i=edge[i].next){  
		int v=edge[i].v;  
		if(v == pre)  
			continue;  
		dfs1(v,u,depth+1);  
		sz[u]+=sz[v];  
		if(sz[son[u]]<sz[v])  
			son[u]=v;  
	}  
}  
void dfs2(int u,int tp){  
	pos[u]=++cnt; top[u]=tp;  
	if(son[u]!=0)  
		dfs2(son[u],top[u]);  
	for(int i=head[u];i!=-1;i=edge[i].next){  
		int v=edge[i].v;  
		if(v == fa[u] || v == son[u])  
			continue;  
		dfs2(v,v);  
	}  
}  
void push_down(int rt){  
	if(segtree[rt].tag){  
		segtree[L(rt)].tag=segtree[R(rt)].tag=segtree[rt].tag;  
		segtree[L(rt)].num=segtree[R(rt)].num=1;  
		segtree[L(rt)].lc=segtree[L(rt)].rc=segtree[rt].lc;  
		segtree[R(rt)].lc=segtree[R(rt)].rc=segtree[rt].lc;  
		segtree[rt].tag=0;  
	}  
}  
void push_up(int rt){  
	segtree[rt].lc=segtree[L(rt)].lc; segtree[rt].rc=segtree[R(rt)].rc;  
	int ans=segtree[L(rt)].num+segtree[R(rt)].num;  
	if(segtree[L(rt)].rc == segtree[R(rt)].lc)  
		ans--;  
	segtree[rt].num=ans;  
}  
void build(int rt,int l,int r){  
	segtree[rt].l=l; segtree[rt].r=r; segtree[rt].num=0;  
	if(l == r)  
		return ;  
	int mid=(l+r)>>1;  
	build(L(rt),l,mid); build(R(rt),mid+1,r);  
}  
void update(int rt,int l,int r,int x){  
	if(segtree[rt].l == l && segtree[rt].r == r){  
		segtree[rt].num=segtree[rt].tag=1;  
		segtree[rt].lc=segtree[rt].rc=x;  
		return ;  
	}  
	push_down(rt);  
	int mid=(segtree[rt].l+segtree[rt].r)>>1;  
	if(r<=mid)  
		update(L(rt),l,r,x);  
	else if(l>mid)  
		update(R(rt),l,r,x);  
	else {  
		update(L(rt),l,mid,x); update(R(rt),mid+1,r,x);  
	}  
	push_up(rt);  
}  
int query(int rt,int l,int r,int L,int R){  
	if(segtree[rt].l == L)  
		Lc=segtree[rt].lc;  
	if(segtree[rt].r == R)  
		Rc=segtree[rt].rc;  
	if(segtree[rt].l == l && segtree[rt].r == r)  
		return segtree[rt].num;  
	push_down(rt);  
	int mid=(segtree[rt].l+segtree[rt].r)>>1;  
	if(r<=mid)  
		return query(L(rt),l,r,L,R);  
	else if(l>mid)  
		return query(R(rt),l,r,L,R);  
	else{  
		int ans=query(L(rt),l,mid,L,R)+query(R(rt),mid+1,r,L,R);  
		if(segtree[L(rt)].rc == segtree[R(rt)].lc)  
			ans--;  
		return ans;  
	}  
	push_up(rt);  
}  
int solve(int u,int v,int id,int c){  
	int ans=0;  
	if(id == 1){  
		while(top[u]!=top[v]){  
			if(dep[top[u]]<dep[top[v]])  
				swap(u,v);  
			update(1,pos[top[u]],pos[u],c);  
			u=fa[top[u]];  
		}  
		if(dep[u]>dep[v])  
			swap(u,v);  
		update(1,pos[u],pos[v],c);  
	}  
	else{
		int ans1=-1,ans2=-1;
		while(top[u]!=top[v]){  
			if(dep[top[u]]<dep[top[v]]){  
				swap(u,v); swap(ans1,ans2);  
			}  
			ans+=query(1,pos[top[u]],pos[u],pos[top[u]],pos[u]);  
			if(Rc == ans1)  
				ans--;  
			ans1=Lc; u=fa[top[u]];  
		}  
		if(dep[u]<dep[v]){  
			swap(u,v); swap(ans1,ans2);  
		}  
		ans+=query(1,pos[v],pos[u],pos[v],pos[u]);  
		if(Rc == ans1)  
			ans--;  
		if(Lc == ans2)  
			ans--;  
	}  
	return ans;  
}  
int main(){  
	while(~scanf("%d%d",&n,&m)){  
		init();  
		for(int i=1;i<=n;i++)  
			scanf("%d",&col[i]);  
		for(int i=1;i<n;i++){  
			int u,v;  
			scanf("%d%d",&u,&v);  
			addedge(u,v); addedge(v,u);  
		}  
		dfs1(1,1,1); dfs2(1,1); build(1,1,n);  
		for(int i=1;i<=n;i++)  
			update(1,pos[i],pos[i],col[i]);  
		while(m--){  
			scanf("%s",str);  
			int u,v;  
			if(str[0] == 'C'){  
				int c;  
				scanf("%d%d%d",&u,&v,&c);  
				solve(u,v,1,c);  
			}  
			else{  
				int u,v;  
				scanf("%d%d",&u,&v);  
				printf("%d\n",solve(u,v,2,0));  
			}  
		}  
	}  
	return 0;  
}