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


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