首页 > 代码库 > hdu1556 Color the ball

hdu1556 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

这题之前做过,有非常多种解法。这次用树状数组做了一下,属于一维的区间问题,也是为后面的二维埋下伏笔。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;
#define maxn 100005
int b[maxn];
int lowbit(int x){
    return x&(-x);
}
void update(int pos,int num)
{
    while(pos<=maxn){
        b[pos]+=num;pos+=lowbit(pos);
    }
}
int getsum(int pos)
{
    int num=0;
    while(pos>0){
        num+=b[pos];pos-=lowbit(pos);
    }
    return num;
}

int main()
{
    int n,m,i,j,c,d;
    while(scanf("%d",&n)!=EOF && n!=0)
    {
        memset(b,0,sizeof(b));
        for(i=1;i<=n;i++){
            scanf("%d%d",&c,&d);
            update(c,1);update(d+1,-1);
        }
        for(i=1;i<=n;i++){
            if(i==n)printf("%d\n",getsum(i));
            else printf("%d ",getsum(i));
        }
        //printf("\n");
    }
    return 0;
}


hdu1556 Color the ball