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;
}