【线段树】SSL

xiaoxiao2021-02-28  15

题意

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

思路

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

代码

#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)); }
转载请注明原文地址: https://www.6miu.com/read-1600157.html

最新回复(0)