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
 
#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;
}

 

posted @ 2019-03-10 22:10  明霞  阅读(132)  评论(0编辑  收藏  举报