POJ 2777 Count Color 线段树

线段树，用int 的30位表示是否涂了某种颜色。

#define maxn 100001
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
using namespace std;

int L,T,O ;//长度，种类，问题数
struct Node{
int mark;
int set;
Node():mark(0),set(0){}
void init(){set=mark=0;}
}node[maxn<<2];
void PushUp(int rt){
node[rt].mark=node[rt<<1].mark|node[rt<<1|1].mark;
}
void PushDown(int rt){
if(node[rt].set){
node[rt<<1].set=node[rt].set;
node[rt<<1|1].set=node[rt].set;
node[rt<<1].mark=node[rt].set;
node[rt<<1|1].mark=node[rt].set;
node[rt].set=0;
}
}
void build(int l,int r,int rt){
if(l==r){
node[rt].mark=1;
node[rt].set=0;
return;
}
int m=(l+r)>>1;
build(ls);
build(rs);
PushUp(rt);
node[rt].set=0;
}
void update(int L,int R,int C ,int l,int r,int rt){
if(L <=l && r <=R){
node[rt].mark=node[rt].set=C;
return;
}
PushDown(rt);
int m=(l+r)>>1;
if(L <=m ) update(L,R,C,ls);
if(R > m ) update(L,R,C,rs);
PushUp(rt);
}
int query(int L,int R,int l,int r,int rt){
if(L <=l &&r <=R){
return node[rt].mark;
}
PushDown(rt);
int ANS=0;
int m=(l+r)>>1;
if(L <=m) ANS|=query(L,R,ls);
if(R > m) ANS|=query(L,R,rs);
return ANS;
}
void SearchAll(int l,int r,int rt){
if(l==r){
OUT3(l,node[rt].mark,node[rt].set);
return;
}
PushDown(rt);
int m=(l+r)>>1;
SearchAll(ls);
SearchAll(rs);
}
void operation(){
char op;int a,b,c;
scanf(" %c%d%d",&op,&a,&b);
if(a>b) {int t=a;a=b;b=t;}
if(op=='C'){
scanf("%d",&c);
update(a,b,1<<(c-1),1,L,1);
}
else{
int D=query(a,b,1,L,1),ANS=0;
for(int i=0;i<T;i++){
ANS+=D&1;
D>>=1;
}
printf("%d\n",ANS);
}
}
int main(void)
{
while(cin>>L>>T>>O ){
build(1,L,1);
for(int i=0;i<O;i++) operation();
}
return 0;
}