SSL_2645 线段树练习题二

题意

与练习题一不同的是，现在给出的每个盒子的颜色都不相同，我们要求出一共能看到几种盒子。

思路

定义一个cover记录颜色，如果有多种颜色我们就把它赋值为-1，之后统计的话我们可以定义一个flag来表示颜色i是否存在，存在的话我们就可以让答案+1。

代码

#include<cstdio>
int n,m,x,y,flag[400001],ans;
struct tr{
    int b,e,cover;
}tree[400001];
void creat(int p,int l,int r)
{
    tree[p].b=l;tree[p].e=r;
    if (r-l>1)
    {
        int m=(l+r)>>1;
        creat(2*p,l,m);
        creat(2*p+1,m,r);
    }
}
void insert(int p,int l,int r,int c)
{
    int m=(tree[p].b+tree[p].e)>>1;
    if (tree[p].b==l&&tree[p].e==r) 
        tree[p].cover=c;
    else
    {
        if (tree[p].cover>=0)//如果此节点已经有颜色了
        {
            tree[p*2].cover=tree[p].cover;//传递给它的子节点
            tree[p*2+1].cover=tree[p].cover;
            tree[p].cover=-1;//多种颜色记录为-1
        }
        if (r<=m) insert(2*p,l,r,c);//左
        else if (l>=m) insert(2*p+1,l,r,c);//右
        else
        {
            insert(2*p,l,m,c);//两边都有
            insert(2*p+1,m,r,c);
        }
    }
}
void count(int p)
{
    if (tree[p].cover>=0) flag[tree[p].cover]=1;//如果这个区间只有一种颜色我们就可以统计
    else if (tree[p].e-tree[p].b>1)//不是叶节点就统计它的子节点
    { 
        count(2*p);
        count(2*p+1);
    }
}
void init()
{
    scanf("%d%d",&n,&m);
    creat(1,1,n);//建树
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        insert(1,x,y,i);//因为每个颜色不同，所以我们用i表示它们的颜色
    }
}
int main()
{
    init();
    count(1);
    for (int i=1;i<=m;i++)
        ans+=flag[i];//统计
    printf("%d",ans);
}