HDU 1556 Color the ball（线段树：区间更新）

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

题意：

N个气球，每次[a,b]之间的气球涂一次色，统计每个气球涂色的次数。

思路：

这道题目用树状数组和线段树都可以，拿这道题来入门一下线段树的区间更新。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn = 1000000 + 5;

int n;
int ans[maxn];

struct node
{
    int l, r;
    int n;
}t[maxn];

int a[maxn];

void build(int l, int r, int o)
{
    t[o].l = l;
    t[o].r = r;
    t[o].n = 0;
    if (l == r) return;
    int mid = (l + r) / 2;
    build(l, mid, 2 * o);
    build(mid + 1, r, 2 * o + 1);
}


void update(int l, int r, int o)
{
    if (t[o].l == l && t[o].r == r)
    {
        t[o].n++;
        return;
    }
    int mid = (t[o].l + t[o].r) / 2;
    if (r <= mid)   update(l, r, 2 * o);
    else if (l > mid)    update(l, r, 2 * o + 1);
    else
    {
        update(l, mid, 2 * o);
        update(mid + 1, r, 2 * o + 1);
    }
}

void add(int o)
{
    for (int i = t[o].l; i <= t[o].r; i++)
    {
        ans[i] += t[o].n;
    }
    if (t[o].l == t[o].r)  return;
    add(2 * o);
    add(2 * o + 1);
}

int main()
{
    //freopen("D:\txt.txt", "r", stdin);
    int x, y;
    while (~scanf("%d", &n) , n)
    {
        memset(ans, 0, sizeof(ans));
        build(1, n, 1);
        for (int i = 1; i <= n; i++)
        {
            scanf("%d%d", &x, &y);
            update(x, y, 1);
        }
        add(1);
        for (int i = 1; i <= n-1; i++)
            cout << ans[i] << " ";
        cout << ans[n] << endl;
    }
}