SSL_2644 线段树练习题一

题意

给出若干盒子,从前面看,求它们不重复的总宽度。

思路

可以用离散化也可以用线段树,线段树会比离散化快。

代码

#include<cstdio>
struct pfa{
    int b,e,v;//b和e代表这个节点所代表的区间的两个范围,v代表此处有没有盒子覆盖
}tree[400001];
int n,m,x,y;
void creat(int p,int l,int r)
{
    tree[p].b=l;tree[p].e=r;
    if (r-l>1)//不是叶节点
    {
        creat(2*p,l,(l+r)/2);//左边
        creat(2*p+1,(l+r)/2,r);//右边
    }
}
void insert(int p,int l,int r)
{
    if (tree[p].v==0)
    {
        int m=(tree[p].e+tree[p].b)/2;
        if (tree[p].b==l&&tree[p].e==r) tree[p].v=1;//如果此节点刚好可以覆盖就记录
        else if (r<=m) insert(2*p,l,r);//左区间的
        else if (l>=m) insert(2*p+1,l,r);//右区间的
        else //两边都有的
        {
            insert(2*p,l,m);
            insert(2*p+1,m,r);
        }
    }

}
void init()
{
    scanf("%d",&n);
    creat(1,1,n);//一开始先建树
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        insert(1,x,y);//插入盒子
    }
}
int count(int p)//统计
{
    if (tree[p].v==1) return tree[p].e-tree[p].b;//如果被覆盖我们就返回它们的长度
    else if (tree[p].e-tree[p].b==1) return 0;//叶节点还没有被记录的话直接返回
    else return count(2*p)+count(2*p+1);//往此节点的两个子节点统计
}
int main()
{
    init();
    printf("%d",count(1));
}

猜你喜欢

转载自blog.csdn.net/ssl_hzb/article/details/80245648