【线段树】SSL_2644 线段树练习题一

题意

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

思路

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

代码

#include
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));
}

你可能感兴趣的:(数据结构)