杭电1556-Color the ball

Color the ball

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5703    Accepted Submission(s): 3049

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<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#define Lson left,mid,n<<1//这里n<<1代表n*2
#define Rson mid+1,right,n<<1|1//这里n<<1|1代表2*n+1
const int MAX=100001;
const int Max=100000<<2;
int s[Max];
int m;
using namespace std;
typedef struct Node
{
    int left;
    int right;
    int value;
};
Node node[Max];
void build_tree(int left,int right,int n)//建树
{
    node[n].left=left;
    node[n].right=right;
    node[n].value=0;//将颜色初始化为0
    if(node[n].left==node[n].right)//如果线段表示的不是left==right就没有到达叶子节点继续下去
    return;
    int mid=(node[n].left+node[n].right)>>1;
    build_tree(Lson);
    build_tree(Rson);
}
void query(int left,int right,int n)
{
    if(node[n].left>=left&&node[n].right<=right)//这里很重要，交上去会不会超时全看这步，这里是用找区间而不是找子节点，找到相应区间就区间加1然后相加时在往下回归到子节点，而如果直接搜到子节点上加1必定超时！
    {
        node[n].value+=1;
        return;
    }
    int mid=(node[n].left+node[n].right)>>1;
    if(right<=mid)//相应区间在左子段树
    query(left,right,n<<1);
    else if(left>mid)//区间在右子段树
    query(left,right,n<<1|1);
    else//区间夹在mid之间即在左子段树和右子段树之间
    {
        query(Lson);
        query(Rson);
    }
}
void sum(int n)
{
    if(node[n].left==node[n].right)//找到子节点
    {
        s[m]=node[n].value;
        m+=1;
        return;
    }
    node[n<<1].value+=node[n].value;//计算每个叶子节点的颜色数且将非叶子节点的颜色加到他的孩子节点上
    node[n<<1|1].value+=node[n].value;
    sum(n<<1);
    sum(n<<1|1);
}
int main()
{
    int n,i,j,a,b;
    while(scanf("%d",&n)&&n)
    {
        build_tree(1,n,1);
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&a,&b);//做这样数据较大的线段树题目建议用scanf和printf输入输出，因为用cin和cout很容易超时
            query(a,b,1);
        }
        m=0;
        sum(1);
        for(i=0;i<m;i++)//输出
        {
            if(i==m-1)
            printf("%d\n",s[i]);
            else
            printf("%d ",s[i]);
        }
    }
    return 0;
}