hdu 1556 Color the ball 【树状数组:区间修改、单点求值】



Color the ball 

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 11813    Accepted Submission(s): 5910


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 <map>
#include <cmath>
#include <vector>
#include <cstdio>
#include <string>
#include <sstream>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100000+5;
int N,a,b,TreeArr[maxn];
int lowbit(int x) { return x&(x^(x-1));}
int Query(int n) {
    //Sum(n) = A1+A2+...+An = Cn+C[n-lowbit(n)]+...
    int sum = 0;
    while(n>0) {
        sum+=TreeArr[n];
        n-=lowbit(n);
    }
    return sum;
}
void Add(int pos,int val) {
    while(pos<=N) {
        TreeArr[pos]+=val;
        pos+=lowbit(pos);
    }
}
void Update(int l,int r,int val) { Add(l,val); Add(r+1,-val); } 
int main() {
   // freopen("input.in","r",stdin);
    while(~scanf("%d",&N) && N) {
        memset(TreeArr,0,sizeof(TreeArr));
        for(int i = 1; i <= N; i++) {
            scanf("%d %d",&a,&b);
            Update(a,b,1);
        }
        for(int i = 1; i <= N; i++) printf("%d%c",Query(i),i==N?'\n':' ');
    }
    return 0;
}


你可能感兴趣的:(树状数组)