HDU 1556 Color the ball

最基础的线段树。。。。

贴个自己写的模板吧

每次insert就是找到匹配的节点 然后update val

此时不再往下update直至叶节点

而是最后查询search的时候再统一从根节点一直加到叶节点。。 因为insert时候没有update匹配节点的子树

所以这样相加其实是没有重复的

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 <stdio.h>
#include <string.h>
#include <algorithm>

#define MAXN 100005
int n;
struct Node
{
    int val, ld, rd;
}t[MAXN<<2];
void build(int pos, int st, int ed)
{
    t[pos].val = 0;
    t[pos].ld = st, t[pos].rd = ed;
    if(st == ed)
        return ;
    int mid = (st+ed)/2 ;
    build(pos*2, st, mid);
    build(pos*2+1, mid+1, ed);
}

void insert(int pos, int st, int ed)
{
    if(t[pos].ld == st && t[pos].rd == ed)
    {
        t[pos].val ++;
        return ;
    }
    int mid = (t[pos].ld + t[pos].rd) /2 ;
    if(ed <= mid)
        insert(pos*2, st, ed);
    else if(st > mid)
        insert(pos*2+1, st, ed);
    else
    {
        insert(pos*2, st, mid);
        insert(pos*2+1, mid+1, ed);
    }
    
}

int search(int pos, int balloon, int sum)
{
    if(t[pos].ld <= balloon && t[pos].rd >= balloon)
        sum += t[pos].val;
    if(t[pos].ld == t[pos].rd)
        return sum;
    int mid = (t[pos].ld + t[pos].rd) /2 ;
    if(balloon <= mid)
        sum = search(pos*2, balloon, sum);
    else
        sum = search(pos*2+1, balloon, sum);
    return sum;
}

int main()
{
    int i, a, b;
    while(scanf("%d", &n), n)
    {
        build(1, 1, n);
        for(i=1; i<=n; i++)
        {
            scanf("%d %d", &a, &b);
            insert(1, a, b);
        }
        for(i=1; i<=n; i++)
        {
            int ans = search(1, i, 0);
            if(i != n)
                printf("%d ", ans);
            else
                printf("%d\n", ans);
        }
        
    }
    return 0;
}