POJ2777 Count Color

题意:

一段区间从1-n的初始颜色为1,每次进行两种操作
1,C a b c 把[a,b]这个区间染成颜色c。
2,P a b查询[a,b]区间内有多少种颜色。

思路:

首先题目保证染色的颜色数少于30种这是关键。我们需要将30种颜色的有无与否都存在线段树的结点上,这样一来每个结点都要存储30个bool值,空间太浪费,而且在计算合并操作的时候有一步30个元素的遍历,大大降低效率。

因为题目保证染色的颜色数少于30种,而30个bool值正好可以压缩在一个int32中(2^{30}=1,073,741,824),利用二进制压缩可以用一个32位的整型完美的存储30种颜色的有无情况。 因为任何一个整数都可以分解成二进制整数,二进制整数的每一位要么是0,要么是1。二进制整数的第i位是1表示存在第i种颜色;反之不存在。 所以我们只需要用一个int来保存一个结点所维护区域的颜色情况一个颜色的lazy标记表示这个结点被完全染成了lazy。所以push_up就变成了或操作,具体看代码。

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 1e5+5;
int sum[maxn<<2],lazy[maxn<<2];
void push_up(int k1){sum[k1] = sum[k1<<1]|sum[k1<<1|1];}
void push_down(int k1,int l,int r){
	if(lazy[k1]){
		int m = (l+r)>>1;
		lazy[k1<<1] = lazy[k1];
		lazy[k1<<1|1] = lazy[k1];
		
		sum[k1<<1] = sum[k1];
		sum[k1<<1|1] = sum[k1];
		lazy[k1] = 0;
	}
}
void build(int k1,int l,int r){
	lazy[k1] = 0;
	if(l == r) {sum[k1] = 1;return;}
	int m = (l+r)>>1;
	build(k1<<1,l,m);
	build(k1<<1|1,m+1,r);
	push_up(k1);	
}
void update(int k1,int l,int r,int L,int R,int v){
	if(L<=l&&r<=R){
		sum[k1] = (1<<(v-1));
		lazy[k1] = 1;
		return;
	}
	push_down(k1,l,r);
	int m = (l+r)>>1;
	if(L<=m) update(k1<<1,l,m,L,R,v);
	if(R>m) update(k1<<1|1,m+1,r,L,R,v);
	push_up(k1);
}

int query(int k1,int l,int r,int L,int R){
	if(L<=l&&r<=R) return sum[k1];	
	push_down(k1,l,r);
	int res = 0;
	int m = (l+r)>>1;
	if(L<=m) res = res | query(k1<<1,l,m,L,R);
	if(R>m) res = res | query(k1<<1|1,m+1,r,L,R);
	return res;
}
int main(){
	int n,t,m,x,y,z;
	char op;
	while(cin>>n>>t>>m){
		build(1,1,n);
		while(m--){
			cin>>op;
			if(op=='C'){
				cin>>x>>y>>z;
				if(x>y) swap(x,y);
				update(1,1,n,x,y,z);
			}
			else{
				cin>>x>>y;
				if(x>y) swap(x,y);
				int ans = 0,tmp = query(1,1,n,x,y);
				//cout<<"Temp"<<tmp<<endl;
				while(tmp) {
					if(tmp&1) ans++;
					tmp>>=1;
				}
				cout<<ans<<'\n';
			}
		}		
	}
	return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值