HDU 1556 Color the ball(树状数组)
今天学习了有点骚气的树状数组(其实学了几天了,今天才理解)
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,输入结束。
当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
#include <cstdio> #include <iostream> #include <string> #include <cstring> #include <cmath> #include <algorithm> using namespace std; int n, a, b, s[100000+8]; int lowbit(int x)//利用机器的反码补码,得出x的二进制末尾处有多少个0 { return x&(-x); } int sum(int x)//算出前x项和 { int ans = 0; while(x>0) { ans += s[x]; x -= lowbit(x); } return ans; } void add(int x, int v)//s数组从x处开始,到最后加上v { while(x<100000+8) { s[x] += v; x += lowbit(x); } } int main() { while(scanf("%d", &n) && n) { fill(s, s+100008, 0); for(int i = 0; i<n; i++) { scanf("%d%d", &a, &b); add(a, 1);//因为从下标a开始一直加到最后一个,所以要使区间[a, b]外的[b+1, maxn]都减去同样的值,保证只在区间[a, b]内增加相应的值 add(b+1, -1); } int flag = 0; for(int i = 1; i <= n; i++) { if(flag)printf(" "); flag = 1; printf("%d", sum(i)); } printf("\n"); } return 0; }