【模板】【线段树】【数据结构】线段树练习题三

题目：

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

---

输入：

一个n，一个l，n行x1,x2,color;

---

输出：

被分成多少段

---

样例输入：

4 20  
10 19 1
2 9 2
5 13 3
15 17 4

---

样例输出：

7

---

思路:

这题用线段树做
首先，题目问我们的是被分成几段，而不是有多少种颜色，所以一段颜色会被另一种一分为二，
像这段粉色，被棕色一分为二，然后我们设cover=-1表示这段区间由多种颜色组成，然后cover>=0表示由单一颜色组成，最后只用判断相邻的颜色是否相同就好。

---

$Code$:

#include<cstdio>
using namespace std;
struct node
{
    int cover;
}tree[1000001];
int x1,y1,l,n;
void insert(int dep,int l,int r,int b,int e,int color)//dep是编号，区间为l~r，查找b~e，color是颜色
//插入操作
{
    if (tree[dep].cover != color)
    {
        int mid=(l + r) >> 1;//中间值
        if (l == b && r == e) tree[dep].cover=color;
        else
        {
            if (tree[dep].cover >= 0)
            {
                tree[dep * 2].cover = tree[dep * 2 + 1].cover=tree[dep].cover;
                tree[dep].cover=-1;//多种
            }
            if (e <= mid) insert(dep * 2,l,mid,b,e,color);else
		    if (b >= mid) insert(dep * 2 + 1,mid,r,b,e,color);else
		     {
                insert(dep * 2,l,mid,b,mid,color);
                insert(dep * 2 + 1,mid,r,mid,e,color);
             }
        }
    }
}
int tongji(int dep,int l,int r,int &x,int &y)
//统计操作
{

    int mid = (l + r) >> 1,xx = 0,yy = 0;
    if (tree[dep].cover >= 0)
    {
        x=y=tree[dep].cover;
        return 1;
    }
    else if (r - l > 1)//多种颜色且区间长度>1
    {
        int ans=tongji(dep*2,l,mid,x,xx)//统计两个孩子
        +tongji(dep*2+1,mid,r,yy,y)-(xx==yy&&xx>=0);//连在一起的相同，减一
        return ans;//返回
       
    }
    return 0;   
}
int main()
{
    scanf("%d",&n);
    scanf("%d",&l);
    int color = 0;
    for (int i = 1;i <= n;i++)
     {
        scanf("%d%d%d",&x1,&y1,&color);
        insert(1,1,l,x1,y1,color);
     }
     int lc = 0,rc = 0;
     printf("%d",tongji(1,1,l,lc,rc));
     return 0;
}