数据结构之线段树 入门HDU1556 color the ball

Color the ball

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

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>
#define mnum 300000
using namespace std;
typedef struct node{
	int data;
	int l,r;
}node;
node tree[mnum];
int n;
int father[mnum/3+2]; //记录单个结点对应的数组标号 
void build(int left,int right,int i)
{
	tree[i].l=left;
	tree[i].r=right;
	tree[i].data=0;
	if(left==right)
	{
		father[left]=i;
		return;
	}
	build(left,(right+left)/2,2*i);
	build(((right+left)/2)+1,right,2*i+1);
}
void insert(int left,int right,int i,int len,int sum)  //对线段树中各点的data（代表color的次数）进行更新，len为待更新区间的长度，sum为已更新的区间的长度！
{	                                                //这个更新的实现是递归实现，但是还可以尝试着用循环来实现，这样时间效率更高！
	int mid=(tree[i].l+tree[i].r)/2;
	if(sum==len)
	return;
	if(left==tree[i].l&&right==tree[i].r)
	{
		tree[i].data++;
		sum+=tree[i].r-tree[i].l;
		return;
	} 
	if(right<mid+1) //如果区间在中值得左边，找左子树 
	{
		insert(left,right,2*i,len,sum);
	}
	else if(left>mid)//如果区间在中值的右边，找右子树 
	{
		insert(left,right,2*i+1,len,sum);
	}
	else //前面两种情况如果都不属于的话，那么待寻找的区间就就部分在了左子树上，且部分在右子树上
	{
		insert(left,mid,2*i,len,sum);
		insert(mid+1,right,2*i+1,len,sum);
	}
} 
int read(int i)
{
	int cnt=0;int t=father[i];
	while(1)
	{
		cnt+=tree[t].data;
		if(t==1)
		return cnt;
		t/=2;
	}
}
int main()
{
	while(cin>>n&&n)
	{	build(1,n,1);
		int a,b;
		for(int i=1;i<=n;i++)
		{
			cin>>a>>b;
			insert(a,b,1,b-a+1,0);
		}
		for(int i=1;i<=n;i++)
		{
			cout<<read(i);
			if(i!=n)
			cout<<" ";
			else cout<<endl;	
		}
	}
}
总结一下，使用线段树后，时间复杂度对每组数据来讲是O(2lgn)