HDU 1556 Color the ball

http://acm.hdu.edu.cn/showproblem.php?pid=1556

 

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 <bits/stdc++.h>
using namespace std;

const int maxn = 1e6 + 10;
int N;
int c[maxn];

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

void add(int i, int val) {
    while(i <= N) {
        c[i] += val;
        i += lowbit(i);
    }
}

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

int main() {
    while(~scanf("%d", &N)) {
        memset(c, 0, sizeof(c));
        if(!N) break;
        for(int i = 0; i < N; i ++) {
            int l, r;
            scanf("%d%d", &l, &r);
            add(l, 1);
            add(r + 1, -1);
        }
        for(int i = 1; i <= N; i ++)
            printf("%d%s", getsum(i), i != N ? " " : "\n");
    }
    return 0;
}

  树状数组

see u

上一篇:codeforces589I


下一篇:zuma_list(使用string类 实现)