hdu1556 Color the ball 简单线段树

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1556

简单的线段树的应用

直接贴代码了：

代码：

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#define maxn 100100
using namespace std;
class node
{
   public:
   int  l;
   int  r;
   int add;
};
int n;
node segTree[maxn*3];
int ans[maxn];
void Build(int num ,int l,int r)
{
   segTree[num].l=l;segTree[num].r=r;
   segTree[num].add=0;
   if(l==r) return ;
   int mid=(l+r)/2;
   Build(num*2,l,mid);
   Build(num*2+1,mid+1,r);
}
void Update(int num,int l,int r)
{
    if(segTree[num].l ==l && segTree[num].r==r)
    {
        segTree[num].add+=1;
        return ;
    }
    int mid=(segTree[num].l + segTree[num].r)/2;
    if(segTree[num].add)
    {
        segTree[num*2].add+=segTree[num].add;
        segTree[num*2+1].add+=segTree[num].add;
        segTree[num].add=0;
    }
    if(r<=mid ) Update(num*2,l,r);
    else if(l>mid) Update(num*2+1,l,r);
    else
        {
            Update(num*2,l,mid);
            Update(num*2+1,mid+1,r);
        }
}
void Answer(int num)
{
    if(segTree[num].l == segTree[num].r)
           {ans[segTree[num].l]=segTree[num].add; return ;}
    if(segTree[num].add)
    {
        segTree[num*2].add+=segTree[num].add;
        segTree[num*2+1].add+=segTree[num].add;
        segTree[num].add=0;
    }
    
   Answer(num*2);
   Answer(num*2+1);
}
int main()
{
    while(scanf("%d",&n)!=EOF && n)
    {
       memset(ans,0,sizeof(ans));
    
       Build(1,1,n);
       int l,r;
       for(int i=1;i<=n;i++)
       {
            scanf("%d%d",&l,&r);
            Update(1,l,r);
       }
       Answer(1);
       cout<<ans[1];
       for(int i=2;i<=n;i++)
          cout<<" "<<ans[i];
          cout<<endl;
    }
    return 0;
}