题目链接
题目大意
L个球,T种颜色,O次操作,初始每个球颜色为1, C为修改a,b区间的颜色为c, P表示询问a,b区间的颜色种数。注意a可能大于b
题目思路
感觉其实是个简单题,看到颜色种类这么少,直接用二进制表示就可以了,好像还可以建30棵线段树,感觉这样也行
代码
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fi first
#define se second
#define debug printf("I am here\n");
using namespace std;
typedef long long ll;
typedef pair<int,pair<int,int> > pii;
const int maxn=1e5+5,mod=998244353,inf=0x3f3f3f3f;
const double eps=1e-10;
int n,m,l,t,o,a,b,c;
int tree[maxn<<2],lazy[maxn<<2];
char ch;
void updown(int node,int l,int r){
if(!lazy[node]) return ;
int mid=(l+r)>>1;
lazy[node<<1]=lazy[node];
lazy[node<<1|1]=lazy[node];
tree[node<<1]=lazy[node];
tree[node<<1|1]=lazy[node];
lazy[node]=0;
}
void build(int node,int l,int r){
if(l==r){
tree[node]=1;
return ;
}
int mid=(l+r)>>1;
build(node<<1,l,mid);
build(node<<1|1,mid+1,r);
tree[node]=tree[node<<1]|tree[node<<1|1];
}
void update(int node,int L,int R,int l,int r,int col){
if(L<=l&&r<=R){
tree[node]=(1<<(col-1));
lazy[node]=(1<<(col-1));
return ;
}
updown(node,l,r);
int mid=(l+r)/2;
if(mid>=L) update(node<<1,L,R,l,mid,col);
if(mid+1<=R) update(node<<1|1,L,R,mid+1,r,col);
tree[node]=tree[node<<1]|tree[node<<1|1];
}
int query(int node,int L,int R,int l,int r){
if(L<=l&&r<=R){
return tree[node];
}
updown(node,l,r);
int mid=(l+r)>>1,sum=0;
if(mid>=L) sum|=query(node<<1,L,R,l,mid);
if(mid<R) sum|=query(node<<1|1,L,R,mid+1,r);
return sum;
}
int cal(ll x){
int cnt=0;
while(x){
if(x&1){
cnt++;
}
x=x>>1;
}
return cnt;
}
signed main(){
while(scanf("%d%d%d",&l,&t,&o)!=-1){
build(1,1,l);
while(o--){
scanf(" %c",&ch);
if(ch=='C'){
scanf("%d%d%d",&a,&b,&c);
if(a>b) swap(a,b);
update(1,a,b,1,l,c);
}else{
scanf("%d%d",&a,&b);
if(a>b) swap(a,b);
int ans=query(1,a,b,1,l);
printf("%d\n",cal(ans));
}
}
}
return 0;
}