染颜色问题
给一个长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值
自己刚开始提交的时候老是超时,后来才发现原来是数组开大了。