题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1556
题目
思路
比较明显的线段树模板题,建立线段树维护区间内气球的涂色次数和,进行区间修改需要打lazy标记,lazy的值就是该区间内的气球涂色且没有下推的次数。
其中PushUp函数用于将结果上推,此题只需将两个子节点的cnt值之和上推即可。PushDown函数用于将lazy标记下推,如果当前结点有lazy标记,只需将其孩子结点的cnt值加上(r-l+1)*lazy即可,就是其孩子结点全部区间上的气球都涂色lazy次。
最后进行区间查询即可得到答案,本题采用线段树结构体的写法。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int maxn = 1e5+10;
struct node{
int l,r;
int cnt;//记录区间内气球涂色的次数和
int lazy;//懒惰标记
}tree[maxn<<2];
//下推标记
void PushDown(int rt)
{
if(tree[rt].lazy){
tree[rt<<1].lazy+=tree[rt].lazy;
tree[rt<<1|1].lazy+=tree[rt].lazy;
tree[rt<<1].cnt+=tree[rt].lazy*(tree[rt<<1].r-tree[rt<<1].l+1);
tree[rt<<1|1].cnt+=tree[rt].lazy*(tree[rt<<1|1].r-tree[rt<<1|1].l+1);
tree[rt].lazy=0;
}
}
//上推标记
void PushUp(int rt)
{
tree[rt].cnt=tree[rt<<1].cnt+tree[rt<<1|1].cnt;
}
//建树
void Build(int l,int r,int rt)
{
tree[rt].l=l;
tree[rt].r=r;
tree[rt].cnt=0;
tree[rt].lazy=0;
if(l==r){
return;
}
int mid=(l+r)>>1;
Build(l,mid,rt<<1);
Build(mid+1,r,rt<<1|1);
PushUp(rt);
}
//区间修改
void Update(int l,int r,int rt)
{
if(l<=tree[rt].l&&tree[rt].r<=r){
tree[rt].cnt+=tree[rt].r-tree[rt].l+1;
tree[rt].lazy+=1;
return;
}
PushDown(rt);
int mid=(tree[rt].l+tree[rt].r)>>1;
if(r<=mid) Update(l,r,rt<<1);
else if(l>mid) Update(l,r,rt<<1|1);
else{
Update(l,mid,rt<<1);
Update(mid+1,r,rt<<1|1);
}
PushUp(rt);
}
//区间查询
int Query(int l,int r,int rt)
{
if(l<=tree[rt].l&&tree[rt].r<=r){
return tree[rt].cnt;
}
PushDown(rt);
int mid=(tree[rt].l+tree[rt].r)>>1;
int ans=0;
if(r<=mid) ans+=Query(l,r,rt<<1);
else if(l>mid) ans+=Query(l,r,rt<<1|1);
else{
ans+=Query(l,mid,rt<<1);
ans+=Query(mid+1,r,rt<<1|1);
}
return ans;
}
int main()
{
// freopen ("input.txt" , "r" , stdin);
// freopen ("output.txt" , "w" , stdout);
int n;
while(1){
scanf("%d",&n);
if(n==0) break;
Build(1,n,1);
for(int i=1;i<=n;i++){
int l,r;
scanf("%d %d",&l,&r);
Update(l,r,1);
}
for(int i=1;i<=n;i++){
printf("%d%c",Query(i,i,1),i==n?'\n':' ');//即查询第i个气球的涂色次数
}
}
return 0;
}