题目:
poj2777
题意:
在一段1 ~ n 的木板进行操作:
C x y v:在区间 x ~ y 中把颜色改成 v
P x y:查询区间 x ~ y 有多少种颜色
思路:
线段树 + lazy + 用位来储存状态
举个例子:
有5个颜色
状态1:颜色1,颜色3
用位表示:00101
状态2:颜色3,颜色4
用位表示:01100
状态3:状态1 和 状态2 的叠加
用位表示:状态1 | 状态2
00101
| 01100
-----------
01101
代码
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#define lson l, mid, pos << 1
#define rson mid + 1, r, pos << 1 | 1
using namespace std;
const int MAXN = 1e5 + 10;
int tree[MAXN<<2];
bool lazy[MAXN<<2];
int bitCount(int x){ //统计 x 有多少个 1
int cnt = 0;
while (x != 0){
x &= (x-1);
cnt++;
}
return cnt;
}
void updata(int p){
tree[p] = tree[p<<1] | tree[p<<1|1];
}
void build(int l, int r, int pos){
if (l == r){
tree[pos] = 1; //初始颜色也是 1
return;
}
int mid = (l + r) >> 1;
build(lson);
build(rson);
updata(pos);
}
void lazyopen(int p){
if (lazy[p]){
tree[p<<1] = tree[p<<1|1] = tree[p];
lazy[p<<1] = lazy[p<<1|1] = true;
lazy[p] = false;
}
}
void add(int s, int e, int v, int l, int r, int pos){
if (s <= l && r <= e){
tree[pos] = v;
lazy[pos] = true;
return;
}
lazyopen(pos);
int mid = (l + r) >> 1;
if (mid >= s)
add(s, e, v, lson);
if (mid < e)
add(s, e, v, rson);
updata(pos);
}
int find(int s, int e, int l, int r, int pos){
if (s <= l && r <= e)
return tree[pos];
lazyopen(pos);
int mid = (l + r) >> 1;
int ans = 0;
if (mid >= s)
ans |= find(s, e, lson);
if (mid < e)
ans |= find(s, e, rson);
return ans;
}
int main(){
// freopen("_in.txt", "r", stdin);
// freopen("_out1.txt", "w", stdout);
int n, tot, m, a, b, c;
char op;
scanf("%d%d%d", &n, &tot, &m);
build(1, n, 1);
while (m--){
scanf(" %c", &op);
if (op == 'C'){
scanf("%d%d%d", &a, &b, &c);
if (a > b) //这是个坑
swap(a, b);
add(a, b, 1<<(c-1), 1, n, 1);
}
else{
scanf("%d%d", &a, &b);
if (a > b) //这是个坑
swap(a, b);
printf("%d\n", bitCount(find(a, b, 1, n, 1)));
}
}
return 0;
}