hdu 1556 Color the ball(樹狀數組)

題目描述:

        N個氣球排成一排,從左到右依次編號爲1,2,3....N.每次給定2個整數a b(a <= b),lele便爲騎上他的“小飛鴿"牌電動車從氣球a開始到氣球b依次給每個氣球塗一次顏色。但是N次以後lele已經忘記了第I個氣球已經塗過幾次顏色了,你能幫他算出每個氣球被塗過幾次顏色嗎?

解題思路:

     這是一道區間更新單點查詢的問題。但是可以轉化成單點更新,區間查詢的問題,比如要給[a,b]區間增加1,那麼可以給a點加1,b+1點-1,那麼在查詢i位置的所塗顏色次數時,就是將[1,i]的值全部相加得到的。所以問題轉化成了一個樹狀數組模板題。

代碼:

#include <iostream>
#include <stdio.h>
#include <cstring>

using namespace std;

int n;
int c[100005];

int lowbit(int x){
    return x&(-x);
}

void updata(int x,int v){
     while(x <= n){
        c[x] += v;
        x += lowbit(x);
     }
}

int query(int x){
     int sum = 0;
     while(x > 0){
        sum += c[x];
        x -=lowbit(x);
     }
     return sum;
}

int main()
{
    int a,b;
    while(true){
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        if(n == 0) break;
        for(int i = 0; i < n; i ++){
            scanf("%d%d",&a,&b);
            updata(a,1);
            updata(b+1,-1);
        }
        for(int i = 1; i < n; i ++){
            printf("%d ",query(i));
        }
        printf("%d\n",query(n));
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章