【SSL2646】线段树练习题3

Description

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

Input

Output

Sample Input

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

Sample Output

7

Hint

把颜色存进数组里，在查询操作时定义两个变量来记录左子树和右子树的颜色，如果它们相等就代表两边是同一种颜色，就把答案加一，若是不同就代表有两种颜色，答案加二

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
struct node{
	long long l,r;
	long long cover;
}f[40000001];
long long c=0,d=0;
long long n,m;
void ins1(long long x,long long l,long long r,long long a,long long b,long long color){
	if(a>b)return;
	if(f[x].cover==color)return;
	if(l==a&&r==b)f[x].cover=color;
	else{
		long long mid=(l+r)>>1;
		if(f[x].cover!=-1){
			f[x*2].cover=f[x*2+1].cover=f[x].cover;
			f[x].cover=-1;
		}
		if(b<=mid)ins1(x*2,l,mid,a,b,color);
		else if(a>mid)ins1(x*2+1,mid+1,r,a,b,color);
		else{
			ins1(x*2,l,mid,a,mid,color);
			ins1(x*2+1,mid+1,r,mid+1,b,color);
		}
	}
}
long long count1(long long x,long long l,long long r,long long &lc,long long &rc){
    long long lt=0,rt=0;
    if(f[x].cover!=-1){
    	lc=rc=f[x].cover;
    	return 1;
    }
    if(l==r)return 1;
    long long mid=(l+r)>>1;
    long long sum=count1(x*2,l,mid,lc,lt)+count1(x*2+1,mid+1,r,rt,rc);
    if(lt==rt)return sum-1;
    else return sum;
}
int main(){
    scanf("%lld%lld",&n,&m);
    for(long long i=1;i<=n;i++){
    	long long x,y,z;
    	scanf("%lld%lld%lld",&x,&y,&z);
    	ins1(1,1,m,x,y,z);
    }
    cout<<count1(1,1,m,c,d);
	return 0;
}