POJ - 2777 Count Color 题解（线段树+bitset试用）

原题

题目大意

给一面墙，有两种操作，C是将A到B涂成C色，P是询问A到B有多少种颜色，初始墙体颜色为1，颜色种类很少($1\le T\le 30$)

题目分析

明显线段树(树状数组好像也行)，然后因为颜色总类很少，我们可以压缩到一个int里面，不过之前遇到了一题bitset优化的毒瘤题不会做，所以用这道题试试bitset
以后不知道会不会写个专题【咕咕咕

构造法

bitset<32> a;//构造一个32位的bitset类型，初始值为0
bitset<4> b(5);//构造一个4位的bitset类型，初始值为0101
bitset<10> c("101");//构造一个10位的bitset类型，初始值为101，前面用0补充
//可以用c标准字符串或者string初始化，注意一定要仅含‘0’和‘1’
std::string s("100111100");
bitset<4> d(s);//只取前面4位，d为1001
bitset<4> e(1026);//只取后四位，e为0010

符号运算

位运算都可以，比较运算里大于小于不能使用
也可以通过下标[]来操作一个数(不推荐，下面有更安全的函数用)

常用函数

bitset<4> b(5);//构造一个4位的bitset类型，初始值为0101
cout << b.count() << endl;//输出1的个数，即2
cout << b.size() << endl;//输出大小，即4
cout << b.any() << endl;//是否有1，这里输出1
cout << b.all() << endl;//是否都是1，这里输出0
cout << b.none() << endl;//是否没有1，这里输出0
cout << b.test(1) << endl;//输出第1位，这里输出0
b.filp();//翻转整个bitset，相当于按位取反，计算过后为1010
b.filp(0);//翻转第0位，计算过后为1011
b.set(2);//将第2位设为1，和b.set(3,1)等价，计算过过后为1111
b.reset();//将b清零，计算过后为0000
b.set();//将b的每一位设为1

代码

#include<cstdio>
#include<bitset>
using std::bitset;

struct
{
    int l,r;
    bitset<32> data,lz;
}tree[500011];
int n,o,t;
void swap(int &x,int &y)
{
    x = x xor y;
    y = y xor x;
    x = x xor y;
}
void build(int i,int l,int r)
{
    tree[i].l = l;tree[i].r = r;
    if (l == r)
    {
        tree[i].data = 2;
        return;
    }
    int mid((l + r) >> 1);
    build(i * 2,l,mid);
    build(i * 2 + 1,mid + 1,r);
    tree[i].data = tree[i * 2].data | tree[i * 2 + 1].data;
}
void push_down(int i)
{
    if (tree[i].lz != 0)
    {
        tree[i * 2].lz = tree[i].lz;
        tree[i * 2 + 1].lz = tree[i].lz;
        tree[i * 2].data = tree[i].lz;
        tree[i * 2 + 1].data = tree[i].lz;
        tree[i].lz.reset();
    }
}
void modify(int i,int l,int r,int k)
{
    if(tree[i].r <= r and tree[i].l >= l)
    {
        tree[i].data.reset();
        tree[i].data.set(k);
        tree[i].lz.reset();
        tree[i].lz.set(k);
        return;
    }
    push_down(i);
    if (tree[i * 2].r >= l) modify(i * 2,l,r,k);
    if (tree[i * 2 + 1].l <= r) modify(i * 2 + 1,l,r,k);
    tree[i].data = tree[i * 2].data | tree[i * 2 + 1].data;
    return;
}

bitset<32> search(int i,int l,int r)
{
    if (tree[i].l >= l and tree[i].r <= r) return tree[i].data;
    if (tree[i].r < l or tree[i].l > r) return 0;
    push_down(i);
    bitset<32> s(0);
    if (tree[i * 2].r >= l) s |= search(i * 2,l,r);
    if (tree[i * 2 + 1].l <= r) s |= search(i * 2 + 1,l,r);
    return s;
}

int main()
{
    scanf("%d%d%d",&n,&t,&o);
    getchar();
    build(1,1,n);
    for (int i = 0;i < o;++i)
    {
        int x,y,z;
        switch (getchar())
        {
        case 'P':
            scanf("%d%d",&x,&y);
            getchar();
            if (x > y) swap(x,y);
            printf("%d\n",search(1,x,y).count());
            break;
        case 'C':
            scanf("%d%d%d",&x,&y,&z);
            getchar();
            if (x > y) swap(x,y);
            modify(1,x,y,z);
            break;
        default:
            break;
        }
    }
    return 0;
}

有件挺坑的事情，我这里没有用第0位，然而我一开始构造的时候写了tree[i].data = 1;WA了，因为这样赋值给bitset会变成1，但本意是涂上第一种颜色，正确的应该是tree[i].data = 2;