树状数组：HDU1556 Color the ball 【区间更新，单点查询]

Color the ball

Time Limit : 9000/3000ms (Java/Other) Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 19 Accepted Submission(s) : 17

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

  
  

Author

8600

Source

HDU 2006-12 Programming Contest

TLE代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,A[100010],C[100010],i,a,b;
int main()
{
	//scanf("%d",&n);
	//if(n==0)break;
	while(scanf("%d",&n),n!=0)
	{
	  	memset(A,0,sizeof(A));
	    memset(C,0,sizeof(C));
	    for(i=0;i<n;i++)
	     {
	        scanf("%d%d",&a,&b);
	        for(int j=a;j<=b;j++)
	        A[j]++;
  	      }
  	      for(i=1;i<=n;i++)
  	      printf("%d ",A[i]);
  	      printf("\n");
	}
	
	return 0;
}	
算法复杂度O(N^2)，这样肯定超时，N=10000的时候N^2已经很勉强了，这个题N=100000呢，铁定超时妥妥的必须从根本上解决，把算法变成O(NlogN)。标准解法线段树。看来来硬的是不行的，看来线段树真的是个省时的好东西。

	
AC代码：

#include<iostream>
#include<cstdio>
using namespace std;
int C[100005],n;//数组开小了会出现运行错误
int Lowbit(int x)
{
	return x&(-x);
}
void Add(int x,int tal)
{
    while(x>0)
    {
    C[x]+=tal;
    x-=	Lowbit(x);	
    }
}
int Sum(int x)
{
   int	  tot=0;
	while(x<=n)
	{
		tot+=C[x];
		x+=Lowbit(x);
	}
	return tot;
}
int main()
{

	while(scanf("%d",&n)&&n)
	{
		int a,b;
		memset(C,0,sizeof(C));
		for(int j=0;j<n;j++)
		{
		scanf("%d%d",&a,&b);
		Add(a-1,-1);
		Add(b,1);	
		}
		
	
	for(int i=1;i<n;i++)
	printf("%d ",Sum(i));
	printf("%d\n",Sum(n));
	}
	return 0;
}

一个空格惹的祸：

如果输出的时候按如下代码：

for(i=0;i<=n;i++)

printf("%d ",Sum(i));

printf("\n");

就会在输出Sum（n）的时候后面多输出一个空格，结果就presentation error;

学长的代码：

#include<cstdio>
#include<cstring>

int tree[110000];
int n;

int lowbit(int x)
{
	return x&-x;
}

void update(int v,int val)
{ // 区间更新 
	while(v>0)
		{
		tree[v]+=val;
		v-=lowbit(v);
	         }
}

int query(int v)
{ // 单点查询 
	int sum=0;
	while(v<=n)
		{
		sum+=tree[v];
		v+=lowbit(v);
	         }
	return sum;
}


int main()
{
	int i,a,b;
	while(~scanf("%d",&n),n)
		{
		memset(tree,0,sizeof(tree));
		for(i=0;i<n;i++)
			{
			scanf("%d%d",&a,&b);
			update(a-1,-1);
			update(b,1);
		         }
		for(i=1;i<=n;i++)
			{
			printf("%d",query(i));
			if(i<n) putchar(' ');
			else putchar('\n');
		         }
	          } 
	return 0;
}