POJ - 2777 Count Color （线段树，区间染色（用二进制表示颜色））

题目连接

 

    题意：
                1 到 L个小木板 染色
                O次操作每次： 
                1）C x y z表示将区间x到y的颜色更改为z
                2）P x y 表示查询区间x到y的颜色种类
    数据范围：
                 1S
                 L 1e5,  O 1e5  颜色T <= 30 
    思路： 
                1）区间更新 + 区间查询  = 1e5 
                2）线段树 ，那线段树上点维护是 [L, R]的颜色数量
                3）因为 颜色数量 < 30 我们就可以把每种颜色看成一个二进制
                    比如颜色 5 就用 10000 表示 
                4）然后线段树上的点 32 位有 几位 1 就是多少个颜色 

AC:

#include<iostream>
#include<math.h>
#include<algorithm>
#include<stdio.h>
#include<cstring>
using namespace std;
typedef long long ll;
const int MAX_LEN = 1e5 + 10;
int N, T, Q;
ll seg_tree[MAX_LEN << 2];
int lazy[MAX_LEN << 2]; 
void push_up(int root) {
	seg_tree[root] = seg_tree[root << 1] | seg_tree[root << 1 | 1];
}
void push_down(int root, int L, int R) {
	if(lazy[root]) {
		lazy[root << 1] = lazy[root];
		lazy[root << 1 | 1] = lazy[root];
		seg_tree[root << 1] = lazy[root]; 
		seg_tree[root << 1 | 1] = lazy[root];	 
		lazy[root] = 0; 
 	}
} 
void build (int root, int L, int R) {
	lazy[root] = 0;
	if (L == R) {
		seg_tree[root] = 1;
		return ; 
	} 
	int mid = (L + R) / 2;
	// [ L, mid] 左子树， [mid + 1, r]右子树 
	build(root << 1, L, mid);
	build(root << 1 | 1, mid + 1, R);
	push_up(root);
	//对与当前的根节点，把当前的根节点的左右子树都算出来后，再更新它的值
	// 沿路回溯， 回溯到点root 时， 都是被 [ L, R] 或者其子区间影响到的点，边回溯边更新 
}
//区间查旬
ll query (int root, int L, int R, int LL ,int RR) {
	if( LL <= L  && R <= RR) return seg_tree[root];
	push_down(root, L, R);	//每次访问都去检查Lazy标记 
	long long Ans = 0;
	int mid = (L + R) >> 1;
	if(LL <= mid) Ans = Ans | query(root << 1, L, mid, LL, RR);
	if(RR > mid) Ans = Ans | query(root << 1 | 1, mid + 1, R, LL, RR);
	return Ans;
}
 
//区间修改 
void update (int root, int L, int R, int LL, int RR, int val) {
	//[LL, RR] 为即将要更新的区间
	if(LL <= L && R <= RR) {
		//cout << L << " " << R << " " << val << endl; 
		lazy[root] = 1 << (val - 1);
		seg_tree[root] = 1 << (val - 1);
		//cout << root << " " << 	lazy[root]  << endl; 
		return ;
	} 
	//更新子树 
	push_down(root, L, R);
	int mid = (L + R) >> 1;
	if(LL <= mid) update(root << 1, L, mid, LL, RR, val);
	if(RR > mid) update(root << 1 | 1, mid + 1, R, LL, RR, val);
	//更新父节点 
	push_up(root);
}
int Solve(int X){
	int Ans = 0;
	while(X) {
		if(X & 1) Ans ++;
		X >>= 1;
	}
	return Ans;
}

int main(){
	//freopen("in.txt", "r", stdin);
	scanf("%d%d%d", &N, &T, &Q);
	char ch[20];
	int L, R, Col;
	build(1, 1, N);
	for(int i = 1; i <= Q; ++i) {
		scanf("%s", ch);
		if(ch[0] == 'C') {
			scanf("%d%d%d", &L, &R, &Col);
			if(L > R) swap(L, R);
			update(1, 1, N, L, R, Col);
		} else {
			scanf("%d%d%d", &L, &R);
			if(L > R) swap(L, R);
			int Ans = query(1, 1, N, L, R);
			printf("%d\n", Solve(Ans));
		}
	}
	return 0;
}