HDOJ 1556 Color the ball 線段樹 ： 成段更新 單點查詢

//HDOJ 1556 Color the ball  線段樹 ： 成段更新 單點查詢
/* 
題目地址：http://acm.hdu.edu.cn/showproblem.php?pid=1556

題目大意：ｎ（ｎ<=100000）個氣球從左到右排列，每次在[a,b]區間上塗一次顏色，
	ｍ次操作後，問第ｉ個氣球一共被塗過幾次顏色。

思路：成段更新
*/

#include<stdio.h>
#include<string.h>

#define N 100005

int sum[N<<2];
int add[N<<2];

void PushUp(int rt){
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void PushDown(int rt,int m){
    if(add[rt]){
        add[rt<<1]   += add[rt];
        add[rt<<1|1] += add[rt];
        sum[rt<<1]   += add[rt] * (m - (m>>1));
        sum[rt<<1|1] += add[rt] * (m>>1);
        add[rt] = 0;
    }
}

void update(int rt,int l,int r,int L,int R){
    if(L <= l && R >= r){
        add[rt] += 1;
        sum[rt] += (r-l+1);
        return;
    }
    PushDown(rt,r-l+1);
    int mid = (r+l)>>1;
    if(L <= mid) update(rt<<1,l,mid,L,R);
    if(R > mid ) update(rt<<1|1,mid+1,r,L,R);
    PushUp(rt);
}

int query(int rt,int l,int r,int L){
    int R = L;
    if(L <= l && R >= r){
        return sum[rt];
    }
    PushDown(rt,r-l+1);
    int mid = (r+l)>>1;
    int res = 0;
    if(L <= mid) res += query(rt<<1,l,mid,L);
    if(R > mid ) res += query(rt<<1|1,mid+1,r,L);
    return res;
}

int main(){
    int n,i;
    int a,b;
    while(scanf("%d",&n),n){
        memset(sum,0,sizeof(sum));
        memset(add,0,sizeof(add));
        for(i = 1; i <= n; ++i){
            scanf("%d %d",&a,&b);
            update(1,1,n,a,b);
        }
        printf("%d",query(1,1,n,1));
        for(i = 2; i <= n; ++i)
            printf(" %d",query(1,1,n,i));
        printf("\n");
    }
    return 0;
}