题意:给你一排气球让你涂色, 有n个操作,表示将[l,r]区间的球涂色,问最后1~n的气球被涂色的次数。
#include<iostream>
#include<cstdio>
#include<cstring>
const int MAXN = 100005;
using namespace std;
struct t
{
int l, r, c;
} tree[MAXN*4];
int lazy[MAXN*4];//数组约4倍大,lazy标记,延迟更新
void pushup(int k)
{ tree[k].c=tree[k<<1].c+tree[k<<1|1].c;
}
void build_tree(int l,int r,int k)
{
tree[k].l=l;
tree[k].r=r;
tree[k].c=0;
if(l==r)
return ;
int mid=(l+r)>>1;
build_tree(l, mid, k<<1);
build_tree(mid+1, r, k<<1|1);
//pushup(k);
}
void pushdown(int k)
{
if(lazy[k])
{
lazy[k<<1]+=lazy[k];
lazy[k<<1|1]+=lazy[k];
tree[k<<1].c+=lazy[k]*(tree[k<<1].r-tree[k<<1].l+1);
tree[k<<1|1].c+=lazy[k]*(tree[k<<1|1].r-tree[k<<1|1].l+1);
lazy[k]=0;
}
return ;
}
void update(int k, int l, int r)
{
if(tree[k].l>=l&&tree[k].r<=r)
{
lazy[k]++;
tree[k].c+=(tree[k].r-tree[k].l+1);
return ;
}
pushdown(k);
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid)//要更新的区间在当前节点的左子树
update(k<<1,l,r);
else if(l>mid)//要更新的区间在当前节点的右子树
update(k<<1|1,l,r);
else//左右子树都有
{
update(k<<1,l,mid);
update(k<<1|1,mid+1,r);
}
pushup(k);
}
int query(int k, int x)
{
if(tree[k].l==tree[k].r)
{
return tree[k].c;
}
pushdown(k);
int ans=0;
int mid = (tree[k].l+tree[k].r)>>1;
if(x<=mid)
ans += query(k<<1, x);
else
ans += query(k<<1|1, x);
return ans;
}
int main()
{
int n;
while(scanf("%d",&n),n)
{
int l, r;
build_tree(1,n,1);
memset(lazy, 0, sizeof(lazy));
for(int i=0; i<n; i++)
{
scanf("%d %d",&l,&r);
update(1,l,r);
}
for(int i=1; i<=n; ++i)
{
printf("%d%c", query(1, i), i==n?'\n':' ');
}
}
return 0;
}
写法二:
#include<iostream>
#include<cstdio>
#include<cstring>
const int MAXN = 100005;
using namespace std;
struct T
{
int l, r, c;
} tree[MAXN*4];
int lazy[MAXN*4];
void build_tree(int l, int r, int k)
{
tree[k].l=l;
tree[k].r=r;
tree[k].c=0;
if(l==r)
return ;
int mid = (tree[k].l+tree[k].r)>>1;
build_tree(l, mid, k<<1);
build_tree(mid+1, r,k<<1|1);
}
void pushup(int k)
{
tree[k].c=tree[k<<1].c+tree[k<<1|1].c;
}
void pushdown(int k)
{
if(lazy[k]!=0)
{
tree[k*2].c+=lazy[k];
tree[k*2+1].c+=lazy[k];
lazy[k*2]+=lazy[k];
lazy[k*2+1]+=lazy[k];
lazy[k]=0;
}
}
void update(int k,int l,int r)
{
if(l<=tree[k].l&&r>=tree[k].r)
{
tree[k].c++;
lazy[k]++;
return ;
}
pushdown(k);
int mid=(tree[k].l+tree[k].r)/2;
if(l<=mid)
update(k*2,l,r);
if(r>mid)
update(k*2+1,l,r);
}
int query(int k,int x)
{
if(tree[k].l==tree[k].r)
{
return tree[k].c;
}
pushdown(k);
int mid=(tree[k].l+tree[k].r)/2;
if(x<=mid)
return query(k*2,x);
else
return query(k*2+1,x);
}
int main()
{
int n;
while(scanf("%d",&n),n)
{
int l, r;
memset(tree, 0, sizeof(tree));
memset(lazy, 0, sizeof(lazy));
build_tree(1, n, 1);
for(int i=0; i<n; i++)
{
scanf("%d %d",&l,&r);
update(1,l,r);
}
for(int i=1; i<=n; ++i)
{
printf("%d%c", query(1, i), i==n?'\n':' ');
}
}
}