POJ 2777

染颜色问题

给一个长X米的木板，分成M段

然后在给N个测试案例

当输入为C A B d的时候就是把从第A块到底B块木板染上颜色d

当输入为P A B 的时候就是查询A到B的木板共染了多少种颜色

输入

2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2

输出

2
1

#include<stdio.h>
#include<string.h>
const int N=500000;
int visit[40];
int color[N];
//void build(int root,int l,int r)
//{
//	color[root]=0;
//	if(l==r)
//	 return;
//	int mid=(l+r)>>1;
//	build(root<<1,l,mid);
//	build(root<<1|1,mid+1,r);
	
//}
void paint(int root,int l,int r,int L,int R,int v)
{
	      if(L<=l&&R>=r)
	      {
	      	 color[root]=v;
	      	 return;
	      }
	      if(color[root])
	      {
	      	 color[root<<1]=color[root<<1|1]=color[root];
	      	 color[root]=0;
	      }
	      int mid=(l+r)>>1;
	    
	     if(L<=mid)
	        paint(root<<1,l,mid,L,R,v);
	      if(R>mid)
	        paint(root<<1|1,mid+1,r,L,R,v);
}
int search(int root,int l,int r,int L,int R)
{
	  int sum=0;
	  if(color[root])
	  {
	  	 if(visit[color[root]]==0)
	  	 {
		   sum++;
		   visit[color[root]]=1;
	  	 }
	  	return sum;   
	  }
	  int mid=(l+r)>>1;
	  if(L<=mid)
	   sum+=search(root<<1,l,mid,L,R);
	  if(R>mid)
	   sum+=search(root<<1|1,mid+1,r,L,R);
	  if(sum==30)
	   return sum;
	return sum;
}
int main()
{
	int x,m,o,W;
	scanf("%d%d%d",&x,&m,&o);
	char c[2];
	int A,B,d;
//	build(1,1,x);
   memset(color,0,N*sizeof(int));
	color[1]=1;
	while(o--)
	{
		scanf("%s",c);
		scanf("%d%d",&A,&B);
		if(A>B)
		  {
		  	W=A;
		  	A=B;
		  	B=W;
		  }
		if(c[0]=='C')
		{
			scanf("%d",&d);
			paint(1,1,x,A,B,d);
		}
		if(c[0]=='P')
		{
			memset(visit,0,40*sizeof(int));
			printf("%d\n",search(1,1,x,A,B));
		}
	}
	return 0;
}

刚开始拿到这道题目的时候，我不知道怎么样去记录木板颜色的种类‘

后来看了别人的解题报告， 才发现，只要用一个数组去标记就可以，当染上了一种颜色后就把这段区间的根节点赋值为color，认然后动态更新，把根节点的子节点赋值，赋值之后把自身变为0

记录某个区间一共染了多少颜色时，先查询该节点是否被染色，然后用visit去标记这个颜色，若该颜色已经加过，则不再累加，若为访问过，则加上该颜色，返回sum值

自己刚开始提交的时候老是超时，后来才发现原来是数组开大了。