poj 2777 Count Color

题目:
poj2777

题意:
在一段1 ~ n 的木板进行操作:
C x y v:在区间 x ~ y 中把颜色改成 v
P x y:查询区间 x ~ y 有多少种颜色

思路:
线段树 + lazy + 用位来储存状态

举个例子:

有5个颜色

状态1:颜色1,颜色3
用位表示:00101

状态2:颜色3,颜色4
用位表示:01100

状态3:状态1 和 状态2 的叠加
用位表示:状态1 | 状态2
			 00101
		   | 01100
		  -----------
		     01101

代码

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#define lson l, mid, pos << 1
#define rson mid + 1, r, pos << 1 | 1
using namespace std;
const int MAXN = 1e5 + 10;
int tree[MAXN<<2];
bool lazy[MAXN<<2];
int bitCount(int x){  //统计 x 有多少个 1
	int cnt = 0;
	while (x != 0){
		x &= (x-1);
		cnt++;
	}
	return cnt;
}
void updata(int p){
	tree[p] = tree[p<<1] | tree[p<<1|1];
} 
void build(int l, int r, int pos){
	if (l == r){
		tree[pos] = 1;  //初始颜色也是 1 
		return;
	}
	int mid = (l + r) >> 1;
	build(lson);
	build(rson);
	updata(pos);
}
void lazyopen(int p){
	if (lazy[p]){
		tree[p<<1] = tree[p<<1|1] = tree[p];
		lazy[p<<1] = lazy[p<<1|1] = true;
		lazy[p] = false;
	}
} 
void add(int s, int e, int v, int l, int r, int pos){
	if (s <= l && r <= e){
		tree[pos] = v;
		lazy[pos] = true;
		return;
	}
	lazyopen(pos);
    int mid = (l + r) >> 1;
	if (mid >= s)
		add(s, e, v, lson);
	if (mid < e)
		add(s, e, v, rson);
	updata(pos);
}
int find(int s, int e, int l, int r, int pos){
	if (s <= l && r <= e)
		return tree[pos];
	lazyopen(pos);
	int mid = (l + r) >> 1;
	int ans = 0;
	if (mid >= s)
		ans |= find(s, e, lson);
	if (mid < e)
		ans |= find(s, e, rson);
	return ans;
}
int main(){
//	freopen("_in.txt", "r", stdin);
//	freopen("_out1.txt", "w", stdout);
	int n, tot, m, a, b, c;
	char op;
	scanf("%d%d%d", &n, &tot, &m);
	build(1, n, 1);
	while (m--){
		scanf(" %c", &op);
		if (op == 'C'){
			scanf("%d%d%d", &a, &b, &c);
			if (a > b)  //这是个坑
				swap(a, b); 
			add(a, b, 1<<(c-1), 1, n, 1);
		}
		else{
			scanf("%d%d", &a, &b);
			if (a > b)  //这是个坑
				swap(a, b); 
			printf("%d\n", bitCount(find(a, b, 1, n, 1)));
		}
	}
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值