POJ2777_Count Color(区间染色+区间查询)

题意：

色板长度为L，L是一个正整数，所以我们可以均匀地将它划分成L块1厘米长的小方格。并从左到右标记为1, 2, … L。现在色板上只有一个颜色，老师告诉阿宝在色板上只能做两件事：1. “C A B C” 指在A到 B 号方格中涂上颜色 C。2. “P A B” 指老师的提问：A到 B号方格中有几种颜色。

思路：

线段树区间染色。颜色有最多30种，用一个int型整数的每一位来表示一种颜色。
初始时全部染色为1，也就是在1的位置染色，00…00010，所以初始化为2；

#include <cstdio>
#include <cstring>
#include <algorithm>
#define lson root<<1, l, mid
#define rson root<<1|1, mid+1, r
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long LL;
const int maxn = 100000+5;
int Tree[maxn<<2], lazy[maxn<<2];

void Stree_build(int root, int l, int r){
	Tree[root] = 2;
	if(l == r) return;
	int mid = (l+r) >> 1;
	Stree_build(lson);
	Stree_build(rson);
}
void push_up(int root){
	Tree[root] = Tree[root<<1] | Tree[root<<1|1]; 
}
void push_down(int root){
	if(lazy[root] == 0) return;
	lazy[root<<1] = lazy[root<<1|1] = lazy[root];
	Tree[root<<1] = Tree[root<<1|1] = (1<<lazy[root]);
	lazy[root] = 0;
}
void update(int la, int rb, int l, int r, int root, int val){
	if(la > r||l > rb) return;
	if(la <= l&&rb >= r){
		Tree[root] = (1<<val);
		lazy[root] = val;
		return;
	}
	push_down(root);
	int mid = (l+r) >> 1;
	if(la <= mid)
		update(la, rb, l, mid, root<<1, val);
	if(rb > mid)
		update(la, rb, mid+1, r, root<<1|1, val);
	push_up(root);
}
void Query(int la, int rb, int l, int r, int root, int& ans){
	if(la > r||l > rb) return;
	if(la <= l&&rb >= r){
		ans|= Tree[root];
		return;
	}
	push_down(root);
	int mid = (l+r) >> 1;
	if(la <= mid)
		Query(la, rb, l, mid, root<<1, ans);
	if(rb > mid)
		Query(la, rb, mid+1, r, root<<1|1, ans);
}

int main()
{
	freopen("in.txt","r",stdin);
	int L, n, m, a,b,c;
	char op[5];
	while(scanf("%d%d%d",&L,&n,&m) == 3){
		memset(lazy, 0, sizeof(lazy));
		Stree_build(1, 1, L);
		for(int i = 0; i < m; ++i){
			scanf("%s",op);
			if(op[0] == 'C'){
				scanf("%d%d%d",&a,&b,&c);
				if(a > b) swap(a, b);
				update(a, b, 1, L, 1, c);
			}
			else{
				scanf("%d%d",&a,&b);
				if(a > b) swap(a, b);
				int ans = 0, x = 0;
				Query(a, b, 1, L, 1, x);
				for(int k = 0; k <= 30; ++k) if(x&(1<<k)) ++ans;
				printf("%d\n", ans); 
			}
		}
	}
	
	return 0;
}