题目链接
题目要求实现区间覆盖修改以及区间数量查询,不难想到为线段树,而需要维护什么值来得到不同数的数量很难想,但是我们注意到颜色的数量最多只有30种,所以对于每一种颜色在一个区间中是否存在,我们可以使用线段树+状态压缩来解决这个问题
首先考虑pushup,这点很简单,只要将两个儿子节点的颜色状态或一下就可以
然后考虑pushdown,此处为颜色覆盖,所以对于每次修改只需要将原先的颜色状态直接覆盖为新的状态即可,包括lazy也是这样,这里注意lazy存的是要覆盖的颜色种类,而改变的时候是要先将1左移lazy个位置然后覆盖
ac代码:
cpp
#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
#define int long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=1e5+10;
struct Tree
{
int l,r;
int sum,lazy;
}tr[N<<2];
int n,m,q;
string op;
int l,r,d;
int lowbit(int x){return x&-x;}
void change(int u,int lazy)
{
tr[u].sum=1<<lazy;
tr[u].lazy=lazy;
}
void pushup(int u)
{
tr[u].sum=tr[u<<1].sum|tr[u<<1|1].sum;
}
void pushdown(int u)
{
if(tr[u].lazy)
{
change(u<<1,tr[u].lazy);
change(u<<1|1,tr[u].lazy);
tr[u].lazy=0;
}
}
void build(int u,int l,int r)
{
if(l==r)tr[u]={l,r,1<<1,0};
else
{
tr[u]={l,r};
int mid=l+r>>1;
build(u<<1,l,mid);
build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(int u,int l,int r,int d)
{
if(tr[u].l>=l&&tr[u].r<=r)
{
change(u,d);
return ;
}
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid)modify(u<<1,l,r,d);
if(r>mid)modify(u<<1|1,l,r,d);
pushup(u);
}
int query(int u,int l,int r)
{
if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
int res=0;
if(l<=mid)res|=query(u<<1,l,r);
if(r>mid)res|=query(u<<1|1,l,r);
return res;
}
void solve()
{
cin>>n>>m>>q;
build(1,1,n);
while(q--)
{
cin>>op>>l>>r;
if(l>r)swap(l,r);
if(op=="C")
{
cin>>d;
modify(1,l,r,d);
}
else
{
int ans=query(1,l,r);
int cnt=0;
while(ans)
{
cnt++;
ans-=lowbit(ans);
}
cout<<cnt<<endl;
}
}
}
signed main()
{
Mirai;
int T=1;
//cin>>T;
while(T--)
{
solve();
}
}