POJ 2777 线段树

链接：

http://poj.org/problem?id=2777

题意：

有L个气球，开始颜色为1，每次给l，r之间的气球染色x，然后询问l，r区间有多少种不同的颜色

题解：

因为颜色最多有30种，所以对这30中颜色状态压缩一下，放在线段树里面，这样就容易更新了

最后只要计算一下query返回值得数有多少个1就行了

代码：

int L, T, O;
int Tree[MAXN << 2];
int Lazy[MAXN << 2];

int BitCount(int n) {
    int c = 0;
    for (c = 0; n; ++c) n &= (n - 1);
    return c;
}

void pushup(int rt) {
    Tree[rt] = Tree[rt << 1] | Tree[rt << 1 | 1];
}

void pushdown(int rt) {
    if (Lazy[rt]) {
        Tree[rt << 1] = Lazy[rt];
        Tree[rt << 1 | 1] = Lazy[rt];
        Lazy[rt << 1] = Lazy[rt];
        Lazy[rt << 1 | 1] = Lazy[rt];
        Lazy[rt] = 0;
    }
}

void build(int l,int r,int rt) {
    if (l == r) {
        Tree[rt] = 1;
        return;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int L, int R, int x, int l, int r, int rt) {
    if (L <= l && r <= R) {
        Tree[rt] = (1 << x);
        Lazy[rt] = (1 << x);
        return;
    }
    pushdown(rt);
    int m = (l + r) >> 1;
    if (L <= m) update(L, R, x, lson);
    if (R > m) update(L, R, x, rson);
    pushup(rt);
}

int query(int L, int R, int l, int r, int rt) {
    if (L <= l && r <= R) return Tree[rt];
    pushdown(rt);
    int m = (l + r) >> 1;
    int ret = 0;
    if (L <= m) ret |= query(L, R, lson);
    if (R > m) ret |= query(L, R, rson);
    return ret;
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> L >> T >> O;
    build(1, L, 1);
    while(O--) {
        char c;
        cin >> c;
        if (c == 'C') {
            int a, b, x;
            cin >> a >> b >> x;
            if (a > b) swap(a, b);
            update(a, b, x - 1, 1, L, 1);
        }
        else {
            int a, b;
            cin >> a >> b;
            if (a > b) swap(a, b);
            cout << BitCount(query(a, b, 1, L, 1)) << endl;
        }
    }
    return 0;
}