线段树练习三 题解

线段树练习三 题解

线段树练习三

题目

给定一条长度为$m$的线段，有$n$个操作，每个操作有3个数字$x$,$y$,$z$表示把区间[$x$,$y$]染成颜色$z$，询问染完色之后，这条长度为$m$的线段一共有几种颜色。规定：线段的颜色可以相同。连续的相同颜色被视作一段。问$x$轴被分成多少段。

---

样例

input
20 //四条,总长度为20
10 19 1
2 9 2
5 13 3
15 17 4

output
7

样例解释

---

数据规模

$N$ <= 10000
$M$ <= 1000000

---

解题思路

搜索覆盖
完全覆盖标为颜色号，多种颜色标为-1
如果与前一段的颜色不一样
段数+1

---

代码

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,x,y,ans,t[2000020],now=-1,c;
void in(int d,int l,int r,int x,int y,int color)
{  
	 int mid=(l+r)/2;
	 if (l==x&&y==r) //完全覆盖
	    t[d]=color;
	    else {
	    	if (t[d]>=0)  //有被完全覆盖（没颜色也算）
	    	{
	    		t[2*d]=t[2*d+1]=t[d];  //子节点拿父节点的颜色
	    		t[d]=-1;  //标记为多种颜色
	    	}
	    	if (x>=mid)
	    	   in(d*2+1,mid,r,x,y,color);
	    	   else if (y<=mid)
	    	           in(d*2,l,mid,x,y,color);
	    	           else {
	    	           	    in(d*2,l,mid,x,mid,color);
	    	           	    in(d*2+1,mid,r,mid,y,color);
	    	           }  //继续找
	    }
}
void tj(int d)
{ 
	 if (t[d]!=-1)  //是被完全覆盖
	 { 
	    if (now!=t[d])  //与前一段不同
	    {
	 	   ans++;   //累加段数
	 	   now=t[d];  //更新当前段的颜色
	    }
	 	return;
	 }
	tj(d*2);
	tj(d*2+1);  //继续搜
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&x,&y,&c);
		in(1,1,m,x,y,c);
	}
	tj(1);
	cout<<ans<<endl;
	return 0;
}