【线段树】SSL_2645 线段树练习题二

题意

与练习题一不同的是，现在给出的每个盒子的颜色都不相同，我们要求出一共能看到几种盒子。

思路

定义一个cover记录颜色，如果有多种颜色我们就把它赋值为-1，之后统计的话我们可以定义一个flag来表示颜色i是否存在，存在的话我们就可以让答案+1。

代码

#include
int n,m,x,y,flag[400001],ans;
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)
{
	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].cover;//传递给它的子节点
			tree[p*2+1].cover=tree[p].cover;
			tree[p].cover=-1;//多种颜色记录为-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) flag[tree[p].cover]=1;//如果这个区间只有一种颜色我们就可以统计
	else if (tree[p].e-tree[p].b>1)//不是叶节点就统计它的子节点
	{ 
		count(2*p);
		count(2*p+1);
	}
}
void init()
{
	scanf("%d%d",&n,&m);
	creat(1,1,n);//建树
	for (int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		insert(1,x,y,i);//因为每个颜色不同，所以我们用i表示它们的颜色
	}
}
int main()
{
	init();
	count(1);
	for (int i=1;i<=m;i++)
		ans+=flag[i];//统计
	printf("%d",ans);
}