bitset线段树

P1558 色板游戏 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
const int N=100010;
int n,m,q;
struct node{
	int l,r,lazy;
	bitset<31>p;
}tr[N*4];
bitset<31>ans;
void pushup(int u){
	tr[u].p=(tr[u*2].p|tr[u*2+1].p);
}
void pushdown(int u,int k){
	tr[u].p.reset();//清空
	tr[u].lazy=k;
	tr[u].p.set(k,1);//第k位变成1
}
void pushdown(int u){//传递懒标记
	if(tr[u].lazy==0) return;
	pushdown(u*2,tr[u].lazy);
	pushdown(u*2+1,tr[u].lazy);
	tr[u].lazy=0;
}
void build(int u,int l,int r){
	tr[u]={l,r};
	tr[u].p.set(1,1);//第1位变成1
	if(l==r) {return;}
	int mid=(l+r)/2;
	build(u*2,l,mid);
	build(u*2+1,mid+1,r);
	pushup(u);
}
void change(int u,int l,int r,int k){
	if(tr[u].l>=l&&tr[u].r<=r){
		tr[u].p.reset();
		tr[u].lazy=k;
		tr[u].p.set(k,1);
		return;
	}
	int mid=(tr[u].l+tr[u].r)/2;
	pushdown(u);
	if(l<=mid) change(u*2,l,r,k);
	if(r>mid) change(u*2+1,l,r,k);
	pushup(u);
}
void query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r){
		ans|=tr[u].p;
		return;
	}
	pushdown(u);
	int mid=(tr[u].l+tr[u].r)/2;
	if(l<=mid) query(u*2,l,r);
	if(r>mid)  query(u*2+1,l,r);
	pushup(u);
	return;
}
void solve()
{
	cin>>n>>m>>q;
	build(1,1,n);
	while(q--){
		char c;int l,r;
		cin>>c>>l>>r;
		if(l>r) swap(l,r);
		if(c=='C'){
			int k;cin>>k;
			change(1,l,r,k);
		}else{
			ans.reset();
			query(1,l,r);//查询1的个数
			cout<<ans.count()<<endl;
		}
	}	
}
signed main()
{
	ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
	solve();
	return 0;
}