大意:给定长度l的木板,m种颜色,k个操作,询问时木板有多少种颜色?
思路:直接给区间染色不容易那么可以间接的化成多少个2相乘的形式,向上给父区间更新的时候直接|,(若相同则不加,否则相加最后求1的个数)。
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<algorithm>
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
using namespace std;
int sum[100000*5],lazy[100000*5],po[40];
char str[10];
void Pushdown(int rt){
if(lazy[rt]){
lazy[rt<<1]=lazy[rt];
lazy[rt<<1|1]=lazy[rt];
sum[rt<<1]=lazy[rt];///因为是直接下方修改颜色所以直接把lazy下放到sum[]上
sum[rt<<1|1]=lazy[rt];
lazy[rt]=0;
}
}
void Pushup(int rt){
sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}
void Updata(int a,int b,int c,int l,int r,int rt){
if(a<=l&&r<=b){
sum[rt]=c;
lazy[rt]=c;
return ;
}
Pushdown(rt);
int mid=(l+r)>>1;
if(a<=mid){
Updata(a,b,c,ls);
}
if(mid<b)
Updata(a,b,c,rs);
Pushup(rt);
}
int Query(int a,int b,int l,int r,int rt){
if(a<=l&&b>=r){
return sum[rt];
}
Pushdown(rt);
int mid=(l+r)>>1;
int ans=0;
if(a<=mid)
ans|=Query(a,b,ls);
if(b>mid)
ans|=Query(a,b,rs);
return ans;
}
int get(int x){
int t=0;
while(x){
if(x&1)
t++;
x>>=1;
}
return t;
}
int main(){
int nu,a,c,i,j,k,n,m,b;
while(~scanf("%d%d%d",&n,&m,&k)){
memset(lazy,0,sizeof(lazy));
for(i=0;i<=31;++i){
po[i]=1<<i;
}
for(i=1;i<=n*5;++i){
sum[i]=1;
}
for(i=0;i<k;++i){
scanf("%s%d%d",str,&a,&b);
if(a>b){
int t=a;
a=b;
b=t;
}
if(!strcmp(str,"C")){
scanf("%d",&c);
Updata(a,b,po[c-1],1,n,1);
}
else if(!strcmp(str,"P")){
int p= Query(a,b,1,n,1);
p=get(p);
printf("%d\n",p);
}
}
//printf("%d\n",n|m);
}
return 0;
}