转载请注明出处: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; }
hdu 1556 Color the ball(树状数组)
时间: 2024-10-07 17:10:50