POJ 2777

染颜色问题

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

然后在给N个测试案例

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

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

输入

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

输出

21

#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值

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