ssl2645-线段树练习2【线段树】

正题

---

题意

一块长m的墙，有n个大小不同的盒子放在前面，求可以看到多少盒子

---

解题思路

用线段树，用cover表示可以看到的颜色，-1表示可以看到多种颜色，然后统计，用find数组去重。

---

代码

#include<cstdio>
using namespace std;
struct xjq{
    int l,r,cover;
}tree[400001];
int n,ll,rr,w,s;
bool flag[100001];
void build(int x,int a,int b)//建树
{
    tree[x].l=a;
    tree[x].r=b;
    if (b-a==1) return;
    else
    {
        int m=(a+b)/2;
        build(x*2,a,m);
        build(x*2+1,m,b);
    }
}
void inster(int x,int a,int b,int c)//插入
{
    if (tree[x].cover==c) return;
    if (tree[x].l==a && tree[x].r==b) //标记颜色
    {
      tree[x].cover=c;return;
    }
    if (tree[x].cover>=0)//下传颜色
    {
        tree[x*2].cover=tree[x].cover;
        tree[x*2+1].cover=tree[x].cover;
        tree[x].cover=-1;
    }
    int m=(tree[x].r+tree[x].l)/2;
    if (b<=m) inster(x*2,a,b,c);
    else if (a>=m) inster(x*2+1,a,b,c);
    else
    {
        inster(x*2,a,m,c);
        inster(x*2+1,m,b,c);
    }
    return;
}
void find(int x)
{
    if (tree[x].cover>=0 && !flag[tree[x].cover])
    {
        s++;
        flag[tree[x].cover]=true;//标记去重
    }
    if (tree[x].cover>=0) return;
    if (tree[x].r-tree[x].l==1) return;
    else 
    {
      find(x*2);
      find(x*2+1);
      return;
    }
}
int main()
{
    scanf("%d",&w);
    scanf("%d",&n);
    build(1,1,w);
    flag[0]=true;
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&ll,&rr);
        inster(1,ll,rr,i);
    }
    s=0;
    find(1);
    printf("%d",s);
}