【树状数组】【HDU1556】Color the ball

【题目描述】

        N个气球排成一排，从左到右依次编号为1,2,3....N。每次给定2个整数a和b(a<=b)，lele便为骑上他的“小飞鸽”牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第i个气球已经涂过几次颜色了，你能帮他算出每个气球被涂过几次颜色吗？

【输入格式】

        每个测试实例第一行为一个整数N(N<=100000)。接下来的N行，每行包括2个整数a b(1<=a<=b<=N)。当N=0，输入结束。

【输出格式】

每个测试实例输出一行，包括N个整数，第i个数代表第i个气球总共被涂色的次数。

【样例输入】                                                【样例输出】

3                                                                                                 1 1 1

1 1                                                                                              3 2 1

2 2
3 3
3
1 1
1 2
1 3
0

 

本题考察的是区间修改加单点查询，线段树代码比较长，我们选择使用更简单的树状数组。

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
int n,tree[400005];
int lowbit(int k)
{
	return k&-k;
}
void add(int x,int k)
{
	while(x<=n)
	{
		tree[x]+=k;
		x+=lowbit(x);
	}
}
int sum(int x)
{
	int ans=0;
	while(x)
	{
		ans+=tree[x];
		x-=lowbit(x);
	}
	return ans;
}
int main()
{
	while(scanf("%d",&n)&&n)
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			add(a,1);
			add(b+1,-1);
		}
		for(int i=1;i<=n;i++)
		{
			printf("%d ",sum(i));
		}
		printf("\n");
	}
}