首页 > 代码库 > hdu 1556 Color the ball（树状数组）

hdu 1556 Color the ball（树状数组）

2024-07-10 02:18:02   229人阅读

转载请注明出处：http://blog.csdn.net/u012860063

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1556


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

解题思路：

树状数组中的每个节点都代表了一段线段区间，每次更新的时候，根据树状数组的特性可以把b以前包含的所有区间都找出来，然后把b之前的区间全部加一次染色次数。然后，再把a之前的区间全部减一次染色次数，这样就修改了树状数组中的[a,b]的区间染色次数，查询每一个点总的染色次数的时候，就可以直接向上统计每个父节点的值，就是包含这个点的所有区间被染色次数，这就是树状数组中向下查询，向上统计的典型应用

树状数组版：

#include <cstdio>  
#include <cstring>  
#define maxn 100047  
int c[maxn]; 
int n,t;
int Lowbit(int x)  // 2^k
{
	return x&(-x);
}
void update(int i, int x)//i点增量为x
{
	while(i > 0)
	{
		c[i] += x;
		i -= Lowbit(i);
	}
}
int sum(int x)//区间求和 [1,x]
{
	int sum=0;
	while(x <= n)
	{
		sum+=c[x];
		x +=Lowbit(x);
	}
	return sum;
}

int main() 
{ 
	int i,j, x, y; 
    while(scanf("%d",&n) && n) 
    { 
        memset(c,0,sizeof(c)); 
        for(i = 1; i <= n; i++)
        { 
            scanf("%d%d",&x,&y);
            update(y,1);//将y以下的区间+1
			update(x-1,-1);//将x以下的区间-1
        }
		for(j = 1; j < n; j++)
		{
			printf("%d ",sum(j));
		}
		printf("%d\n",sum(n));
    } 
    return 0; 
}


非树状数组版：（其实也是运用了树状数组版的思路）

#include <cstdio>
#include <cstring>
int main()
{
	int n, i, j, a[100047], x, y;
	while(scanf("%d",&n) && n)
	{
		memset(a,0,sizeof(a));
		for(i = 1; i <= n; i++)
		{
			scanf("%d%d",&x,&y);
			a[x]++,a[y+1]--;
		}
		int sum = a[1];
		printf("%d",a[1]);
		for(i = 2; i <= n; i++)
		{
			sum+=a[i];
			printf(" %d",sum);
		}
		printf("\n");
	}
	return 0;
}


联系
我们