Luogu P1558 色板游戏

闲扯

一道线段树好题裸题

顺便吐槽一下出题人的数据。。。

最开始没看到 \(A\) 可能大于 \(B\) ，结果 \(WA\) 了。。

题面

题面

Solution

看范围， \(T\leq30\) ，发现可以用二进制表示是否包含每一种颜料，接下来的事就好办了。

对于操作 \(1\) ，是一个区间覆盖，直接将 \(s\) 变成 \(1<<(C-1)\) ，表示这个区间现在只含有 \(C\) 这种颜料，懒标记换一下。

下放也是一样的操作。

向上统计的时候直接用左区间的 \(s\) 或上右区间的 \(s\) 即 \(T[cur].s=T[lc].s\mid T[rc].s\) 。

每一次查询时找到表示这段区间的 \(s\) ，然后直接查看有包含几个 \(1\) 即可。

Code

#include<bits/stdc++.h>
#define del(a,i) memset(a,i,sizeof(a))
#define ll long long
#define inl inline
#define il inl void
#define it inl int
#define ill inl ll
#define re register
#define ri re int
#define rl re ll
#define mid ((l+r)>>1)
#define lowbit(x) (x&(-x))
#define INF 0x3f3f3f3f
using namespace std;
template<class T>il read(T &x){
	int f=1;char k=getchar();x=0;
	for(;k>'9'||k<'0';k=getchar()) if(k=='-') f=-1;
	for(;k>='0'&&k<='9';k=getchar()) x=(x<<3)+(x<<1)+k-'0';
	x*=f;
}
template<class T>il print(T x){
	if(x/10) print(x/10);
	putchar(x%10+'0');
}
ll mul(ll a,ll b,ll mod){long double c=1.;return (a*b-(ll)(c*a*b/mod)*mod)%mod;}
it qpow(int x,int m,int mod){
	int res=1,bas=x%mod;
	while(m){
		if(m&1) res=(res*bas)%mod;
		bas=(bas*bas)%mod,m>>=1;
	}
	return res%mod;
}
const int MAXN = 1e5+5;
int n,m,k,x,y,z;
char opt;
#define lc (cur<<1)
#define rc (cur<<1|1)
struct Seg_Tree{
	int tag,s;
}T[MAXN<<2];
il pushup(int cur){T[cur].s=T[lc].s|T[rc].s;}
il pushdown(int cur,int l,int r){
	if(!T[cur].tag) return ;
	T[lc].tag=T[rc].tag=T[cur].tag;
	T[lc].s=T[rc].s=1<<(T[cur].tag-1);
	T[cur].tag=0;
}
il build_tree(int cur,int l,int r){
	T[cur].s=1;
	if(l==r) return ;
	build_tree(lc,l,mid),build_tree(rc,mid+1,r);
}
it query(int cur,int l,int r,int L,int R){
	if(l>=L&&r<=R) return T[cur].s;
	pushdown(cur,l,r);ri res=0;
	if(mid>=L) res|=query(lc,l,mid,L,R);
	if(R>mid) res|=query(rc,mid+1,r,L,R);
	return res;
}
il updata(int cur,int l,int r,int L,int R,int k){
	if(l>=L&&r<=R) T[cur].s=1<<(k-1),T[cur].tag=k;
	else{
		pushdown(cur,l,r);
		if(mid>=L) updata(lc,l,mid,L,R,k);
		if(R>mid) updata(rc,mid+1,r,L,R,k);
		pushup(cur);
	}
}
it count(int x){
	ri res=0;
	while(x){
		if(x&1) res++;
		x>>=1;
	}
	return res;
}
int main()
{
//	freopen("testdata.in","r",stdin);
//	freopen("1.out","w",stdout);
	read(n),read(k),read(m);
	build_tree(1,1,n);
	for(ri i=1;i<=m;++i){
		cin>>opt;read(x),read(y);
		if(x>y) swap(x,y);
		if(opt=='C') read(z),updata(1,1,n,x,y,z);
		else{
			ri tmp=query(1,1,n,x,y);
			print(count(tmp)),puts("");
		}
	}
	return 0;
}

总结

蒟蒻颓了一晚上还是做点题吧。。

马上 \(NOIP\) 了虽然它已经死了，感觉好慌呀 \(qwq\) 。

这样的模板题要追求速度，同时还要保证正确率，不然考场上一慌就 \(GG\) 。