hdu——1556Color the ball(树状数组)

Color the ball

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 15310    Accepted Submission(s): 7629


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

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

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

Sample Input
   
   
   
   
3 1 1 2 2 3 3 3 1 1 1 2 1 3 0
 

Sample Output
   
   
   
   
1 1 1 3 2 1
 

Author
8600
 

Source
HDU 2006-12 Programming Contest
 

Recommend
LL

 

树状数组在改变一个数据时整个数组的值都会改变  在这道题目中  需要求某一个气球涂过的次数 假如结果(a,b)  那么需要进行add(a,1);然后进行add(b+1,-1)来抵消对b以后的气球的影响 

#include<iostream>
#include<string>
#include<cstring>
using namespace std;
long long sum;
int tree[100010];
long long lowbit(long long n)
{
	return n&-n;
}
void add(long long n,long long m)
{
	while(n<=sum)
	{
		tree[n]+=m;
		n+=lowbit(n);
	}
}
long long getsum(long long n)
{
	long long h=0;
	while(n>0)
	{
		h+=tree[n];
		n-=lowbit(n);
	}
	return h;
}
int main()
{
	long long n,m,i,a[100010];
	while(cin>>sum&&sum)
	{
		memset(a,0,sizeof(a));
		memset(tree,0,sizeof(tree));
		for(i=1;i<=sum;i++)
		{
			cin>>n>>m;
			add(n,1);
			add(m+1,-1);
		}
		for(i=1;i<=sum;i++)
		{
			if(i==1)cout<<getsum(i);//只需要求某个点的次数 不需要将前面几个气球的和相加
			else cout<<" "<<getsum(i);
		}
		cout<<endl;
	}
	return 0;
}

你可能感兴趣的:(C++,树状数组)