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，输入结束。

 

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

 

#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("
");
    }
    return 0;
}