Color the ball(线段树)

Color the ball
Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 23799 Accepted Submission(s): 11570

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

using namespace std;

#define MAXN 440000
//Maxn 代表结点个数最大值
struct Node
{
    int left;
    int right;
    long long cnt;//涂的次数
}tree[MAXN];
long long add[MAXN];//延迟标记
long long father[MAXN];//记录叶子节点的下标
void buildtree(int rt,int l,int r)
{
    tree[rt].left = l;
    tree[rt].right = r;
    tree[rt].cnt = 0;
    if(l==r)//到达叶子节点 递归结束
    {
        return ;
    }
    //递归左子树 区间范围是[l,(l+r)/2];
    buildtree(rt*2,l,(l+r)/2);
    //递归创建右子树 区间范围是[(l+r)/2 +1.r]
    buildtree(rt*2+1,(l+r)/2+1,r);
}
void PushDown(int rt,int num)
{
    if(add[rt])//如果延迟标记不为0 呢么就继续往下传递延迟
    {
        add[rt*2] += add[rt];
        add[rt*2 +1] += add[rt];
        tree[rt*2].cnt += add[rt] * (num - num/2);//其中左区间在更新cnt (num - num/2) 为左区间的个数
        tree[rt*2+1].cnt += add[rt] * (num / 2);//其中右区间在更新cnt
        add[rt] =0;//并将当前的延迟标记置0
    }
}
void update(int rt,int l,int r,int c)
{
    if(l== tree[rt].left && r == tree[rt].right)
    {
        add[rt] ++;
        tree[rt].cnt +=(r - l +1);//这里一定是c*(r-l+1) 不是add[rt] *(r - l+1) 
        return;
    }
    if(tree[rt].left==tree[rt].right) return; //不更新叶子节点

    PushDown(rt,tree[rt].right - tree[rt].left +1);//设置延迟标记
    //如果这个待更新的区间
    int mid = (tree[rt].left + tree[rt].right)/2;
    if(r<= mid)
        update(rt*2,l,r,c);//更新左子树
    else if(l > mid)
        update(rt*2+1,l,r,c);//更新右子树
    else
    {
        update(rt*2,l,mid,c);
        update(rt*2+1,mid+1,r,c);
    }
    tree[rt].cnt = (tree[rt*2].cnt + tree[rt*2 +1].cnt);//递归之后进行pushup
}
long long query(int rt,int l,int r)//查询区间
{
    if(l== tree[rt].left && r == tree[rt].right)
    {
        return tree[rt].cnt;
    }

    PushDown(rt,tree[rt].right - tree[rt].left +1);
    int mid = (tree[rt].left + tree[rt].right)/2;
    long long ans =0;
    if(r<= mid)
      ans =  query(rt *2, l, r);//继续在左区间进行更新
    else if(l > mid )
    {
       ans = query(rt *2+1,l, r);//继续在右区间进行更新
    }
    else
    {
        ans += query(rt *2, l, mid);//位于两个区间
        ans += query(rt *2+1, mid+1, r);
    }
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n),n)
    {
        buildtree(1,1,n);//建树
        for(int i =1;i<=n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            update(1,a,b,1);//数目加一
        }
        printf("%lld",query(1,1,1));
        for(int i = 2;i<=n;i++)
        {
            printf(" %lld",query(1,i,i));
        }
        printf("\n");
        memset(add,0,sizeof(add));
        memset(tree,0,sizeof(tree));
    }
    return 0;
}