HDU1556 Color the ball（线段树模板+lazy标记）

题目链接：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;
}