HDU 1556 Color the ball（树状数组）

一时失言乱红尘 2021-12-22 14:01 224阅读 0赞

今天学习了有点骚气的树状数组（其实学了几天了，今天才理解）

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 ][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("\n");
        }
        return 0;
    }

转载于:https://www.cnblogs.com/RootVount/p/10507611.html

[HDU 2006-12 Programming Contest]: http://acm.hdu.edu.cn/search.php?field=problem&key=HDU+2006-12+Programming+Contest+&source=1&searchmode=source