poj 2777 Count Color 线段树区间更新

题目大意：

给你两种操作：

1、给一段连续区间涂颜色。

2、统计某一区间有多少种不同的颜色。

分析：

线段树区间更新，区间查询。略坑的一点是要用二进制优化，开始用的vector TLE到死。。。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <set>
#include <algorithm>
#include <cstring>
#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r

using namespace std;
const int maxn=1e5+5;
int data[maxn*4];
int cnt[maxn*4];

void pushdown(int k){
    if(cnt[k]){
        data[k<<1]=data[k<<1|1]=data[k];
        cnt[k<<1]=cnt[k<<1|1]=1;cnt[k]=0;
    }
}
void update(int k,int l,int r,int a,int b,int v){
    if(a>r||b<l)return ;
    if(a<=l&&r<=b){cnt[k]=1;data[k]=1<<v;return ;}
    pushdown(k);
    int mid=l+r>>1;
    update(lson,a,b,v);
    update(rson,a,b,v);
    data[k]=data[k<<1|1]|data[k<<1];
}
int query(int k,int l,int r,int a,int b){
    if(a>r||b<l){return 0;}
    if(a<=l&&r<=b){return data[k];}
    pushdown(k);
    int mid=l+r>>1;
    int t1=query(lson,a,b);
    int t2=query(rson,a,b);
    return t1|t2;
}

inline void read(int &z)  {
    bool sig = false;
    char ch = getchar();
    while (ch<'0' || ch>'9') {ch=='-'?sig=true:false;ch = getchar();}
    int x = ch - '0';
    while ((ch = getchar()) >= '0'&&ch <= '9') x = x * 10 + ch - '0';
    z = sig ? -x : x;
}

inline void out(int x){
    if ( x > 9 ) out ( x / 10 );
    putchar ( x % 10 + '0');
}

int main()
{
    int L,T,O;
    while(~scanf("%d%d%d",&L,&T,&O)){
        int haha=min(L*8,maxn*4);
        for(int i=0;i<haha;i++){data[i]=1<<1;}
        fill(cnt,cnt+haha,0);
        for(int i=1;i<=O;i++){
            int a,b,c;
            char p;scanf("\n%c",&p);
            if(p=='C'){
                read(a),read(b),read(c);
                if(a>b)swap(a,b);
                update(1,1,L,a,b,c);
            }
            else{
                read(a),read(b);
                if(a>b)swap(a,b);
                int t=query(1,1,L,a,b);
                int ans=0;
                for(int i=0;i<31;i++){
                    if(t&(1<<i))ans++;
                }
                out(ans);puts("");
            }
        }
    }
    return 0;
}