题目传送门: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)));
}
}
}