【线段树】SSL_2646 线段树练习题三

题意

在一条线上染上几种颜色，求这些颜色把这条线分为几段。

思路

这道题的插入算法和练习题二差不多，只不过统计的时候要改一下。

代码

#include<cstdio>
int n,m,x,y,ans,z,f;
struct tr{
	int b,e,cover;
}tree[400001];
void creat(int p,int l,int r)
{
	tree[p].b=l;
	tree[p].e=r;
	if (r-l>1)
	{
		int m=(l+r)>>1;
		creat(2*p,l,m);
		creat(2*p+1,m,r);
	}
}
void insert(int p,int l,int r,int c)
{
	if (tree[p].cover!=c)//如果颜色不重复才进行操作，下面的和练习题二的差不多，就不做解释了
	{
		int m=(tree[p].b+tree[p].e)>>1;
		if (tree[p].b==l&&tree[p].e==r) tree[p].cover=c;
		else
		{
			if (tree[p].cover>=0)
			{
				tree[p*2].cover=tree[p*2+1].cover=tree[p].cover;
				tree[p].cover=-1;
			}
			if (r<=m) insert(2*p,l,r,c);
			else if (l>=m) insert(2*p+1,l,r,c);
			else
			{
				insert(2*p,l,m,c);
				insert(2*p+1,m,r,c);
			}
		}
	}
}
void count(int p)
{
	if (tree[p].cover>=0) 
	{
		if (tree[p].cover!=f)//用f记录我们下一段的颜色，如果当前的和它不相同，我们就统计答案，并且更新f
		{
			ans++;
			f=tree[p].cover;
		}
		return;
	}
	if (tree[p].e-tree[p].b>1)//不是叶节点就访问子节点
	{
		count(p*2);
		count(p*2+1);
	}
}
void init()
{
	scanf("%d%d",&n,&m);
    creat(1,1,m);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		insert(1,x,y,z);
	}
}
int main()
{
	init();
	count(1);
	printf("%d",ans);
}