线段树（压位）luogu P1558色板游戏

Posted 2020-12-28 eric-walker

题目背景

阿宝上学了，今天老师拿来了一块很长的涂色板。

题目描述

色板长度为L，L是一个正整数，所以我们可以均匀地将它划分成L块1厘米长的小方格。并从左到右标记为1, 2, ... L。

现在色板上只有一个颜色，老师告诉阿宝在色板上只能做两件事：

1. "C A B C" 指在A到 B 号方格中涂上颜色 C。
2. "P A B" 指老师的提问：A到 B号方格中有几种颜色。

学校的颜料盒中一共有 T 种颜料。为简便起见，我们把他们标记为 1, 2, ... T. 开始时色板上原有的颜色就为1号色。 面对如此复杂的问题，阿宝向你求助，你能帮助他吗？

输入输出格式

输入格式：

 

第一行有3个整数 L (1 <= L <= 100000), T (1 <= T <= 30) 和 O (1 <= O <= 100000)。 在这里O表示事件数。
接下来 O 行, 每行以 "C A B C" 或 "P A B" 得形式表示所要做的事情(这里 A, B, C 为整数, 可能A> B，这样的话需要你交换A和B)

 

输出格式：

 

对于老师的提问，做出相应的回答。每行一个整数。

 

输入输出样例

输入样例#1：

2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2

输出样例#1：

2
1


简单的线段树维护压位（当然你也可以开30颗线段树）

code：

#include<iostream>
#include<cstdio>
using namespace std;
inline int ls(int x){return x<<1;}
inline int rs(int x){return x<<1|1;}
const int maxn=1e5+5;
struct FFF{
    int l,r;
	int sum;
	int tag;
    int mid(){return l+r>>1;}
    int len(){return r-l+1;}
}t[maxn<<2];
int U,T;
int n,m;
void up(int o){
	t[o].sum=t[ls(o)].sum|t[rs(o)].sum;
}
void build(int l=1,int r=n,int o=1){
    t[o].l=l;t[o].r=r;
	t[o].tag=0;
    if(l==r){
        t[o].sum=1;
        return;
    }
    int mid=l+r>>1;
    build(l,mid,ls(o));
    build(mid+1,r,rs(o));
    up(o);
}
void down(int o){
	int &v=t[o].tag;
	if(v){
		for(int i=0;i<=1;++i){
			t[ls(o)|i].tag=v;
			t[ls(o)|i].sum=v;
		}
		v=0;
	}
}
void add(int l,int r,int pos,int o=1){
    if(l<=t[o].l&&t[o].r<=r){
		t[o].tag=(1<<(pos-1));
		t[o].sum=(1<<(pos-1));
        return;
    }
    int mid=t[o].mid();
    down(o);
    if(l<=mid)add(l,r,pos,ls(o));
    if(r>mid)add(l,r,pos,rs(o));
    up(o);
}
int getsum(int l,int r,int o=1){
    if(l<=t[o].l&&t[o].r<=r){
        return t[o].sum;
    }
    int ans=0;
    int mid=t[o].mid();
    down(o);
    if(l<=mid)ans|=getsum(l,r,ls(o));
    if(r>mid)ans|=getsum(l,r,rs(o));
    return ans;
}
inline int bcount(unsigned int x){
	int ans=0;
	for(;x;x>>=1)if(x&1)++ans;
	return ans;
}
signed main(){
    cin>>n>>T>>m;
	U=(1<<(T))-1;
	build();
	while(m--){
		char op[3];
		int l,r,x;
		scanf("%s",op);
		cin>>l>>r;
		if(r<l)r^=l^=r^=l;
		if(op[0]==‘C‘){cin>>x;add(l,r,x);}
		if(op[0]==‘P‘){
			int ans=getsum(l,r);
			cout<<bcount(ans)<<endl;
		}
	}
    return 0;
}