POJ 2777 Count Color题解（线段树染色）

题目链接

题目大意

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;
}