POJ 2777 Count Color (线段树成段更新+二进制思维)

题目链接：http://poj.org/problem?id=2777

题意是有L个单位长的画板，T种颜色，O个操作。画板初始化为颜色1。操作C讲l到r单位之间的颜色变为c，操作P查询l到r单位之间的颜色有几种。

很明显的线段树成段更新，但是查询却不好弄。经过提醒，发现颜色的种类最多不超过30种，所以我们用二进制的思维解决这个问题，颜色1可以用二进制的1表示，同理，颜色2用二进制的10表示，3用100，...。假设有一个区间有颜色2和颜色3，那么区间的值为二进制的110(十进制为6)。那我们就把一个区间的颜色种类表示为（左孩子的值‘|’右孩子的值）。

然后就是一个线段树的成段更新。

有个坑点是这个题目的l可能比r大...

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 275000;
struct segtree {
    int l , r , val , lazy;
}T[MAXN << 3];
//获取颜色的种类
int get(int n) {
    int cont = 0;
    while(n) {
        if(n & 1) 
            cont++;
        n >>= 1;
    }
    return cont;
}

void init(int p , int l , int r) {
    int mid = (l + r) >> 1;
    T[p].l = l , T[p].r = r , T[p].lazy = 0;
    if(l == r) {
        T[p].val = 1;
        return ;
    }
    init(p << 1 , l , mid);
    init((p << 1)|1 , mid + 1 , r);
    T[p].val = (T[p << 1].val | T[(p << 1)|1].val);
}

void updata(int p , int l , int r , int val) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(l == T[p].l && T[p].r == r) {
        T[p].val = val;
        T[p].lazy = val;
        return ;
    }
    if(T[p].lazy) {
        T[p << 1].val = T[(p << 1)|1].val = T[p].lazy;
        T[p << 1].lazy = T[(p << 1)|1].lazy = T[p].lazy;
        T[p].lazy = 0;
    }
    if(r <= mid) {
        updata(p << 1 , l , r , val);
    }
    else if(l > mid) {
        updata((p << 1)|1 , l , r , val);
    }
    else {
        updata(p << 1 , l , mid , val);
        updata((p << 1)|1 , mid + 1 , r , val);
    }
    T[p].val = (T[p << 1].val | T[(p << 1)|1].val);
}
//返回颜色种类的二进制对应的十进制的值
int query(int p , int l , int r) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(T[p].l == l && T[p].r == r) {
        return T[p].val;
    }
    if(T[p].lazy) {
        T[p << 1].val = T[(p << 1)|1].val = T[p].lazy;
        T[p << 1].lazy = T[(p << 1)|1].lazy = T[p].lazy;
        T[p].lazy = 0;
    }
    if(r <= mid) {
        return query(p << 1 , l , r);
    }
    else if(l > mid) {
        return query((p << 1)|1 , l , r);
    }
    else {
        return query(p << 1 , l , mid) | query((p << 1)|1 , mid + 1 , r);
    }
}

int main()
{
    int L , t , n , l , r , c;
    char str[5];
    while(~scanf("%d %d %d" , &L , &t , &n)) {
        init(1 , 1 , L);
        while(n--) {
            scanf("%s" , str);
            if(str[0] == 'C') {
                scanf("%d %d %d" , &l , &r , &c);
                int temp = l + r;
                l = min(l , r);
                r = temp - l;
                updata(1 , l , r , (1 << (c - 1)));
            }
            else {
                scanf("%d %d" , &l , &r);
                int temp = l + r;
                l = min(l , r);
                r = temp - l;
                int res = get(query(1 , l , r));
                printf("%d
" , res);
            }
        }
    }
}