HDOJ 1556 Color the ball(线段树—区间所有值更新+逐个查找值)



Color the ball

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

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<cstdio>
int ans;
struct node
{
	int left,right,count;
}num[500000];//数组小了，超时了一发 

void build(int left,int right,int cnt)
{
	int mid;
	num[cnt].left=left;
	num[cnt].right=right;
	num[cnt].count=0;
	if(left==right)
		return ;
	mid=(left+right)>>1;
	build(left,mid,cnt*2);
	build(mid+1,right,cnt*2+1);
}

void update(int left,int right,int cnt)
{
	int mid;
	if(left==num[cnt].left&&right==num[cnt].right)
	{
		num[cnt].count++;
		return ;
	}
	if(num[cnt].left==num[cnt].right)
		return ;
	mid=(num[cnt].left+num[cnt].right)>>1;
	if(right<=mid)
		update(left,right,cnt*2);
	else if(left>mid)
		update(left,right,cnt*2+1);
	else
	{
		update(left,mid,cnt*2);
		update(mid+1,right,cnt*2+1);
	}
}

void query(int pos,int cnt)
{
	int mid;
	ans+=num[cnt].count;
	if(num[cnt].left==num[cnt].right)
		return ;
	mid=(num[cnt].left+num[cnt].right)>>1;
	if(pos>mid)
		query(pos,cnt*2+1);
	else
		query(pos,cnt*2);
}

int main()
{
	int n,a,b,i;
	while(scanf("%d",&n)&&n)
	{
		build(1,n,1);
		int m=n;
		while(m--)
		{
			scanf("%d%d",&a,&b);
			update(a,b,1);
		}
		for(i=1;i<=n;++i)
		{
			ans=0;
			query(i,1);
			if(i>1)
				printf(" ");
			printf("%d",ans);
		}
		printf("\n");
	}
	return 0;
}