HDU 1556-Color the ball【ツリー配列】

1697 ワード

Color the ball
Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 11088    Accepted Submission(s): 5546
Problem Description
N個の風船が一列に並んでいて、左から右に1,2,3.....N.が2個の整数a b(a<=b)を与えるたびに、leleは彼の「小飛鳩」ブランドの電気自動車に乗って風船aから風船bまで順番に各風船に色を塗った.しかしN回以降leleはI個目の風船が何回か色を塗ったことを忘れてしまったので、風船ごとに何回塗られたか算出してもらえますか?
 
Input
各試験例の第1の挙動は整数N(N<=100000)であり、次のN行は、各行に2つの整数a b(1<=a<=b<=N)を含む.
N=0の場合、入力は終了します.
 
Output
各試験例は、N個の整数を含む1行を出力し、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 <stdio.h>
#include <string.h>

int c[100010];
int n;
int lowbit(int x){
    return x&(-x);
}

void add (int pos,int val){
    while(pos<=n){
        c[pos]+=val;
        pos+=lowbit(pos);
    }
    return ;
}

int sum(int pos){
    int s=0;
    while(pos>0){
        s+=c[pos];
        pos-=lowbit(pos);
    }
    return s;
}
int main (){
    int a,b,i;
    while(scanf("%d",&n),n){
        memset(c,0,sizeof(c));
        for(i=1;i<=n;++i){
            scanf("%d%d",&a,&b);
            add(a,1);
            add(b+1,-1);//  
        }
        for(i=1;i<n;++i)
            printf("%d ",sum(i));
        printf("%d
",sum(n)); } return 0; }