luogu P1558 色板游戏

题目传送门：https://www.luogu.org/problemnew/show/P1558

题意：

有n个位置，每一个位置都有一个颜色，初始颜色为1。现在有两个操作：

[1]将x~y区间的颜色改为z；

[2]统计x~y区间不同颜色数的个数。

思路：

考虑用一颗线段树来维护这一些操作，发现统计颜色好像做不到，就可以想到颜色的范围<31，所以考虑状态压缩。用一个30位的数来表示当前的颜色，如1表示为1，2表示为10，以此类推，用一个异或搞一个改段求段即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
	int n,m,q,len=0;
	struct node{int d,lazy,l,r,lc,rc;} tr[200010];
void build(int l,int r)
{
	int now=++len,mid=(l+r)>>1;
	tr[now].l=l;
	tr[now].r=r;
	tr[now].d=1;
	tr[now].lazy=0;
	tr[now].lc=tr[now].rc=-1;
	if(l<r)
	{
		tr[now].lc=len+1; build(l,mid);
		tr[now].rc=len+1; build(mid+1,r);
	}
}
void updata(int x)
{
	if(tr[x].lazy)
	{
		int lc=tr[x].lc,rc=tr[x].rc;
		tr[lc].lazy=tr[rc].lazy=tr[x].lazy;
		tr[lc].d=tr[rc].d=tr[x].lazy;
		tr[x].lazy=0;
	}
}
void change(int now,int l,int r,int k)
{
	updata(now);
	if(l==tr[now].l&&r==tr[now].r)
	{
		tr[now].d=k;
		tr[now].lazy=k;
		return;
	}
	int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)>>1;
	if(r<=mid) change(lc,l,r,k);
	else if(l>mid) change(rc,l,r,k);
	else change(lc,l,mid,k),change(rc,mid+1,r,k);
	tr[now].d=tr[lc].d|tr[rc].d;
}
int findsum(int now,int l,int r)
{
	updata(now);
	if(l==tr[now].l&&r==tr[now].r) return tr[now].d;
	int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)>>1;
	if(r<=mid) return findsum(lc,l,r);
	else if(l>mid) return findsum(rc,l,r);
	else return findsum(lc,l,mid)|findsum(rc,mid+1,r);
}
int work(int x)
{
	int sum=0;
	while(x>0)
	{
		sum+=(x&1);
		x>>=1;
	}
	return sum;
}
int main()
{
	char s[5];
	int x,y,z;
	scanf("%d %d %d",&n,&m,&q);
	build(1,n);
	for(int i=1;i<=q;i++)
	{
		scanf("%s",s+1);
		if(s[1]=='C')
		{
			scanf("%d %d %d",&x,&y,&z);
			if(x>y) swap(x,y);
			change(1,x,y,1<<(z-1));
		}
		else
		{
			scanf("%d %d",&x,&y);
			if(x>y) swap(x,y);
			printf("%d\n",work(findsum(1,x,y)));
		}
	}
}