题意:
一段区间从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中(),利用二进制压缩可以用一个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;
}