BZOJ 2243 [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]之间。

Source

第一轮day1

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

树链剖分~

对于每一个区间我们记录c[k].l c[k].r c[k].lc c[k].rc c[k].num，分别表示左右位置，最左面节点的颜色，最右面节点的颜色以及区间内段数。所以每次合并时判断两区间相交处是否颜色相同即可。剩余的部分就和普通的树链剖分一样了~

注意dep[1]要设为1，防止lca出错。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

int n,m,x,y,z,a[100001],fi[100001],w[200001],ne[200001],cnt,dep[100001],son[100001],id[100001],siz[100001],fa[100001][17],top[100001];
char s[2];

struct node{
	int l,r,lc,rc,num,tag;
}c[100001<<2];

int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}

void add(int u,int v)
{
	w[++cnt]=v;ne[cnt]=fi[u];fi[u]=cnt;
	w[++cnt]=u;ne[cnt]=fi[v];fi[v]=cnt;
}

void dfs1(int u)
{
	for(int i=1;i<=16;i++) fa[u][i]=fa[fa[u][i-1]][i-1];
	siz[u]=1;
	for(int i=fi[u];i;i=ne[i])
	  if(w[i]!=fa[u][0])
	  {
	  	fa[w[i]][0]=u;dep[w[i]]=dep[u]+1;dfs1(w[i]);siz[u]+=siz[w[i]];
	  	if(!son[u] || siz[son[u]]<siz[w[i]]) son[u]=w[i];
	  }
}

void dfs2(int u,int la)
{
	id[u]=++cnt;top[u]=la;
	if(son[u]) dfs2(son[u],la);
	for(int i=fi[u];i;i=ne[i])
	  if(w[i]!=fa[u][0] && w[i]!=son[u]) dfs2(w[i],w[i]);
}

int lca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=16;~i;i--) if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];
	if(u==v) return u;
	for(int i=16;~i;i--) if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}

void update(int k)
{
	int l=k<<1,r=k<<1|1;
	c[k].lc=c[l].lc;c[k].rc=c[r].rc;
	c[k].num=c[l].num+c[r].num;
	if(c[l].rc==c[r].lc) c[k].num--;
}

void pushdown(int k)
{
	int l=c[k].l,r=c[k].r;
	if(c[k].tag==-1 || l==r) return;
	c[k<<1].tag=c[k<<1|1].tag=c[k].tag;
	c[k<<1].lc=c[k<<1|1].lc=c[k<<1].rc=c[k<<1|1].rc=c[k].tag;c[k].tag=-1;
	c[k<<1].num=c[k<<1|1].num=1;
}

void build(int l,int r,int k)
{
	c[k].l=l;c[k].r=r;c[k].tag=-1;c[k].num=1;
	if(l==r) return;
	int mid=l+r>>1;
	build(l,mid,k<<1);build(mid+1,r,k<<1|1);
}

void chan(int k,int u,int v,int col)
{
	pushdown(k);
	int l=c[k].l,r=c[k].r,mid=l+r>>1;
	if(l>=u && r<=v)
	{
		c[k].lc=c[k].rc=c[k].tag=col;c[k].num=1;return;
	}
	if(mid>=u) chan(k<<1,u,v,col);
	if(mid<v) chan(k<<1|1,u,v,col);
	update(k);
}

int getcol(int k,int u)
{
	pushdown(k);
	int l=c[k].l,r=c[k].r,mid=l+r>>1;
	if(l==r) return c[k].lc;
	if(mid>=u) return getcol(k<<1,u);
	return getcol(k<<1|1,u);
}

void chancol(int u,int v,int col)
{
	int x=top[u],y=top[v];
	while(x!=y)
	{
		chan(1,id[x],id[u],col);u=fa[x][0];x=top[u];
	}
	chan(1,id[v],id[u],col);
}

int cal(int k,int u,int v)
{
	pushdown(k);
	int l=c[k].l,r=c[k].r,mid=l+r>>1,now;
	if(l>=u && r<=v) return c[k].num;
	if(mid>=v) return cal(k<<1,u,v);
	if(mid<u) return cal(k<<1|1,u,v);
	now=cal(k<<1,u,v)+cal(k<<1|1,u,v);
	if(c[k<<1].rc==c[k<<1|1].lc) now--;
	return now;
}

int getnum(int u,int v)
{
	int x=top[u],y=top[v],now=0;
	while(x!=y)
	{
		now+=cal(1,id[x],id[u]);
		if(getcol(1,id[x])==getcol(1,id[fa[x][0]])) now--;
		u=fa[x][0];x=top[u];
	}
	return now+cal(1,id[v],id[u]);
}

int main()
{
	n=read();m=read();
	for(int i=1;i<=n;i++) a[i]=read();
	for(int i=1;i<n;i++) x=read(),y=read(),add(x,y);
	cnt=0;dep[1]=1;dfs1(1);dfs2(1,1);build(1,n,1);
	for(int i=1;i<=n;i++) chan(1,id[i],id[i],a[i]);
	while(m--)
	{
		scanf("%s",s);x=read();y=read();
		if(s[0]=='C')
		{
			int k=lca(x,y);
			z=read();chancol(x,k,z);chancol(y,k,z);
		}
		else
		{
			int k=lca(x,y);
			printf("%d\n",getnum(x,k)+getnum(y,k)-1);
		}
	}
	return 0;
}