【BZOJ 2243】 [SDOI2011]染色

2243: [SDOI2011]染色

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 2291   Solved: 890
[ Submit][ Status]

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

树链剖分+线段树的lazy标记传递

分别说一下对于两个操作我的做法：

1.C(Paint):

维护一个变量modi，表示被修改成了多少，若modi=-1则表示没有被修改。

只要在线段树中找到要修改的区间，把modi赋值即可；修改之后要Push_up

2.Q(Query):

首先找到lca，然后对于u和v所在的重链进行相同的计算：

只要top[u]!=top[lca]就加上当前点到top[u]的颜色段数量。

注意要记录p，表示前一次计算中的最高点是谁。那么如果当前计算的最低点的颜色等于p的颜色，ans--。

top[u]=top[lca]后，再加上u到lca的颜色段数量，对p进行同样的处理。

最后从u和从v的计算结果相加再减1（因为lca被重复算了两次）就是答案。

<见代码中注释>

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <cmath>
#define pb push_back
#define N 100005
using namespace std;
struct segtree
{
	int modi,l,r,lc,rc,cs;
}a[N*3];
vector<int> v[N];
int siz[N],n,m,son[N],fa[N],dep[N],top[N],id[N],co[N],cc[N],num;
void dfs1(int x,int f,int de)
{
	siz[x]=1;
	son[x]=0;
	fa[x]=f;
	dep[x]=de;
	for (int i=0;i<v[x].size();i++)
		if (v[x][i]!=fa[x])
		{
			int u=v[x][i];
			dfs1(u,x,de+1);
			siz[x]+=siz[u];
			if (siz[son[x]]<siz[u])
				son[x]=u;
		}
}
void dfs2(int x,int tp)
{
	top[x]=tp;
	id[x]=++num;
	if (son[x]) dfs2(son[x],tp);
	for (int i=0;i<v[x].size();i++)
	{
		int u=v[x][i];
		if (u==fa[x]||u==son[x]) continue;
		dfs2(u,u);
	}
}
void push_down(int x)
{
	if (!x) return;
	if (a[x].modi!=-1)
	{
		a[x].lc=a[x].rc=a[x].modi;  
		a[x].cs=1;
		if (a[x].l==a[x].r)  //注意对叶子结点的特殊处理
		{
			a[x].modi=-1;
			return;
		}
		a[x<<1].modi=a[x<<1|1].modi=a[x].modi;
		a[x].modi=-1;
	}
}
void push_up(int x)
{
	if (!x||a[x].l==a[x].r) return;
	push_down(x<<1),push_down(x<<1|1);   //一定要先下传
	a[x].lc=a[x<<1].lc,a[x].rc=a[x<<1|1].rc;
	a[x].cs=a[x<<1].cs+a[x<<1|1].cs;
	if (a[x].cs>1&&a[x<<1].rc==a[x<<1|1].lc) a[x].cs--;
}
void Build(int x,int l,int r)
{
	a[x].l=l,a[x].r=r,a[x].modi=-1;
	if (l==r) 
	{
		a[x].lc=a[x].rc=cc[l];
		a[x].cs=1;
		return;
	}
	int m=(l+r)>>1;
	Build(x<<1,l,m);
	Build(x<<1|1,m+1,r);
	push_up(x);
}
void Modify(int x,int l,int r,int c)
{
	if (a[x].l>=l&&a[x].r<=r)
	{
		a[x].modi=c;
		return;
	}
	push_down(x);
	int m=(a[x].l+a[x].r)>>1;
	if (l<=m) Modify(x<<1,l,r,c);
	if (r>m) Modify(x<<1|1,l,r,c);
	push_up(x);
}
void Paint(int u,int v,int c)
{
	int tp1=top[u],tp2=top[v];
	while (tp1!=tp2)
	{
		if (dep[tp1]<dep[tp2]) swap(tp1,tp2),swap(u,v);
		Modify(1,id[tp1],id[u],c);
		u=fa[tp1];
		tp1=top[u];
	}
	if (u==v)
	{
		Modify(1,id[u],id[u],c);
		return;
	}
	if (dep[u]>dep[v]) swap(u,v);
	Modify(1,id[u],id[v],c);
	return;
}
int Count(int x,int l,int r)
{
	push_down(x);
	if (a[x].l==l&&a[x].r==r) return a[x].cs;  
	int m=(a[x].l+a[x].r)>>1;
	if (r<=m) return Count(x<<1,l,r);
	if (l>m) return Count(x<<1|1,l,r);
	int ans=Count(x<<1,l,m)+Count(x<<1|1,m+1,r);
	if (a[x<<1].rc==a[x<<1|1].lc&&ans>1) ans--;
	return ans;
}
int Getcolor(int x,int p)
{
	push_down(x);
	if (a[x].l==p&&a[x].r==p) return a[x].lc;
	int m=(a[x].l+a[x].r)>>1;
	if (p<=m) return Getcolor(x<<1,p);
	else return Getcolor(x<<1|1,p);
}
int Query(int u,int v)
{
	int uu=u,vv=v;
	int ans=0,tp1=top[u],tp2=top[v];
	while (tp1!=tp2)            //找lca
	{
		if (dep[tp1]<dep[tp2]) swap(tp1,tp2),swap(uu,vv);
		uu=fa[tp1];
		tp1=top[uu];
	}
	int lca=tp1,h;       //这里的h是lca,lca是top[lca]
	if (dep[uu]>dep[vv]) h=vv;
	else h=uu;
	tp1=top[u],tp2=top[v];
	int p=0;
	while (tp1!=lca)   //从u开始算
	{
		ans=ans+Count(1,id[tp1],id[u]);
		if (p&&Getcolor(1,p)==Getcolor(1,id[u])) ans--;
		p=id[tp1];
		u=fa[tp1];
		tp1=top[u];
	}	
	ans+=Count(1,id[h],id[u]);
	if (p&&Getcolor(1,p)==Getcolor(1,id[u])) ans--;
	p=0;
	while (tp2!=lca)   //从v开始算
	{
		ans=ans+Count(1,id[tp2],id[v]);
		if (p&&Getcolor(1,p)==Getcolor(1,id[v])) ans--;
		p=id[tp2];
		v=fa[tp2];
		tp2=top[v];
	}
	ans+=Count(1,id[h],id[v]);
	if (p&&Getcolor(1,p)==Getcolor(1,id[v])) ans--;
	ans--;       //lca这个点被计算了两次
	return ans;
}
int main()
{
        scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
		scanf("%d",&co[i]);
	for (int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		v[x].pb(y),v[y].pb(x);
	}
	num=0;
	dfs1(1,0,1);
	dfs2(1,1);
	for (int i=1;i<=n;i++)
		cc[id[i]]=co[i];
	Build(1,1,n);
	while (m--)
	{
		char s[10];
		int x,y,z;
		scanf("%s",s);
		if (s[0]=='C')
		{
			scanf("%d%d%d",&x,&y,&z);
			Paint(x,y,z);
		}
		else
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",Query(x,y));
		}
	}
	return 0;
}

感悟：

1.CE了半个小时没查出来错。

查了一下说是函数名和编译器模板重复了，把所有函数过程的首字母大写之后就过了。

以后所有函数过程的首字母都大写吧！

2.WA:query中我算成了u和v都到达top[lca]的颜色段数了，其实到lca就行了。。

还是得细心啊。。。

3.对于有lazy标记的题目在Push_down和Push_up要特殊考虑叶子结点没有儿子以及a[0]的问题！