POJ 2777Count Color (线段树区间更新 + 位运算压缩)

大意：给定长度l的木板，m种颜色,k个操作，询问时木板有多少种颜色？

思路：直接给区间染色不容易那么可以间接的化成多少个2相乘的形式，向上给父区间更新的时候直接|，（若相同则不加，否则相加最后求1的个数）。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<algorithm>
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1

using namespace std;
int sum[100000*5],lazy[100000*5],po[40];
char str[10];

void Pushdown(int rt){
    if(lazy[rt]){
        lazy[rt<<1]=lazy[rt];
        lazy[rt<<1|1]=lazy[rt];

        sum[rt<<1]=lazy[rt];///因为是直接下方修改颜色所以直接把lazy下放到sum[]上
        sum[rt<<1|1]=lazy[rt];
        lazy[rt]=0;
    }
}

void Pushup(int rt){
    sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}

void Updata(int a,int b,int c,int l,int r,int rt){
    if(a<=l&&r<=b){
        sum[rt]=c;
        lazy[rt]=c;
        return ;
    }
    Pushdown(rt);
    int mid=(l+r)>>1;
    if(a<=mid){
        Updata(a,b,c,ls);
    }
    if(mid<b)
        Updata(a,b,c,rs);
    Pushup(rt);
}
int Query(int a,int b,int l,int r,int rt){
    if(a<=l&&b>=r){
        return  sum[rt];
    }
    Pushdown(rt);

    int mid=(l+r)>>1;
    int ans=0;
    if(a<=mid)
          ans|=Query(a,b,ls);
    if(b>mid)
          ans|=Query(a,b,rs);
    return ans;
}
int get(int x){
    int t=0;
    while(x){
        if(x&1)
            t++;
        x>>=1;
    }
    return t;
}
int main(){
    int nu,a,c,i,j,k,n,m,b;
    while(~scanf("%d%d%d",&n,&m,&k)){
        memset(lazy,0,sizeof(lazy));
        for(i=0;i<=31;++i){
            po[i]=1<<i;
        }
        for(i=1;i<=n*5;++i){
            sum[i]=1;
        }
        for(i=0;i<k;++i){
            scanf("%s%d%d",str,&a,&b);
            if(a>b){
                int t=a;
                a=b;
                b=t;
            }
            if(!strcmp(str,"C")){
                scanf("%d",&c);
                Updata(a,b,po[c-1],1,n,1);
            }
            else if(!strcmp(str,"P")){
               int p= Query(a,b,1,n,1);
               p=get(p);
               printf("%d\n",p);
            }
        }
        //printf("%d\n",n|m);


    }
    return 0;
}