线段树真是一种优雅的解决方案。
将30种颜色用二进制编码,然后用线段树每个节点维护颜色的种类。
void push_up(int rt) {
tree[rt] = tree[rt<<1] | tree[rt<<1|1];
}
巧妙的位运算使得通过每个二进制位即可知道在这个区域内一共有多少种颜色。
另外用lazy数组延迟标记以优化算法速度。
需要注意的是输入的a,b需要比较一下,因为可能会存在a > b的情况。
#include <cstdio>
#include <algorithm>
#define lchild rt << 1, l, m
#define rchild rt << 1 | 1, m + 1, r
const int maxn = 100000 + 5;
const int sz = 300000 + 5;
int tree[sz];
int lazy[sz];
int n;
void push_up(int rt) {
tree[rt] = tree[rt<<1] | tree[rt<<1|1];
}
void push_down(int rt) {
tree[rt<<1] = tree[rt<<1|1] = 1 << (lazy[rt] - 1);
lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];
lazy[rt] = 0;
}
void build(int rt = 1, int l = 1, int r = n) {
if (l == r) {
tree[rt] = 1;
lazy[rt] = 0;
return;
}
int m = (l + r) >> 1;
build(lchild);
build(rchild);
push_up(rt);
}
void update(int L, int R, int color, int rt = 1, int l = 1, int r = n) {
if (L <= l && r <= R) {
tree[rt] = 1 << (color - 1);
lazy[rt] = color;
return;
}
if (lazy[rt]) {
push_down(rt);
}
int m = (l + r) >> 1;
if (L <= m) {
update(L, R, color, lchild);
}
if (R > m) {
update(L, R, color, rchild);
}
push_up(rt);
}
int query(int L, int R, int rt = 1, int l = 1, int r = n) {
if (L <= l && r <= R)
return tree[rt];
if (lazy[rt])
push_down(rt);
int m = (l + r) >> 1, ret = 0;
if (L <= m)
ret |= query(L, R, lchild);
if (R > m)
ret |= query(L, R, rchild);
return ret;
}
int main(int argc, char const *argv[]) {
int l, t, o;
scanf("%d%d%d%*c", &l, &t, &o);
n = l;
build();
int a, b, c;
for (int i = 0; i < o; i++) {
char ch = getchar();
if (ch == 'C') {
scanf("%d%d%d%*c", &a, &b, &c);
if (a > b) {
std::swap(a, b);
}
update(a, b, c);
} else {
scanf("%d%d%*c", &a, &b);
if (a > b) {
std::swap(a, b);
}
int color = query(a, b), cnt = 0;
for (int i = 0; i < t; i++) {
if (color & (1 << i)) {
cnt++;
}
}
printf("%d\n", cnt);
}
}
return 0;
}