线段树练习三 题解
题目
给定一条长度为 m m m的线段,有 n n n个操作,每个操作有3个数字 x x x, y y y, z z z表示把区间[ x x x, y y y]染成颜色 z z z,询问染完色之后,这条长度为 m m m的线段一共有几种颜色。规定:线段的颜色可以相同。连续的相同颜色被视作一段。问 x x x轴被分成多少段。
样例
input
20 //四条,总长度为20
10 19 1
2 9 2
5 13 3
15 17 4
output
7
样例解释
数据规模
N
N
N <= 10000
M
M
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;
}