洛谷P1558 色板游戏

题目背景

阿宝上学了，今天老师拿来了一块很长的涂色板。

题目描述

色板长度为L，L是一个正整数，所以我们可以均匀地将它划分成L块1厘米长的小方格。并从左到右标记为1, 2, ... L。现在色板上只有一个颜色，老师告诉阿宝在色板上只能做两件事：1. "C A B C" 指在A到 B 号方格中涂上颜色 C。2. "P A B" 指老师的提问：A到 B号方格中有几种颜色。学校的颜料盒中一共有 T 种颜料。为简便起见，我们把他们标记为 1, 2, ... T. 开始时色板上原有的颜色就为1号色。 面对如此复杂的问题，阿宝向你求助，你能帮助他吗？

输入输出格式

输入格式：

 

第一行有3个整数 L (1 <= L <= 100000), T (1 <= T <= 30) 和 O (1 <= O <= 100000). 在这里O表示事件数， 接下来 O 行, 每行以 "C A B C" 或 "P A B" 得形式表示所要做的事情(这里 A, B, C 为整数, 可能A> B)

 

输出格式：

 

对于老师的提问，做出相应的回答。每行一个整数。

 

输入输出样例

输入样例#1：

2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2

输出样例#1：

2
1
分析：这道题肯定是要用线段树来做，但是要怎么保存信息呢？保存颜色数肯定不对，我们要保存拥有哪些颜色，但是用数组又不太好，观察发现颜色的个数非常少，所以可以转化为二进制来做，每次pushup就用|操作就好了.

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

int L,T,O,c[1000010],flag[1000010];

void pushup(int o)
{
    c[o] = c[o * 2] | c[o * 2 + 1];
}

void pushdown(int o)
{
    if (flag[o])
    {
        flag[o * 2] = flag[o * 2 + 1] = flag[o];
        c[o * 2] = c[o * 2 + 1] = flag[o];
        flag[o] = 0;
    }
}

void build(int l,int r,int o)
{
    if (l == r)
    {
        c[o] = 1;
        return;
    }
    int mid = (l + r) >> 1;
    build(l,mid,o * 2);
    build(mid + 1,r,o * 2 + 1);
    pushup(o);
}

void update(int l,int r,int o,int x,int y,int v)
{
    if (x <= l && r <= y)
    {
        c[o] = v;
        flag[o] = v;
        return;
    }
    pushdown(o);
    int mid = (l + r) >> 1;
    if (x <= mid)
    update(l,mid,o * 2,x,y,v);
    if (y > mid)
    update(mid + 1,r,o * 2 + 1,x,y,v);
    pushup(o);
}

int query(int l,int r,int o,int x,int y)
{
    if (x <= l && r <= y)
    return c[o];
    pushdown(o);
    int cnt = 0,mid = (l + r) >> 1;
    if (x <= mid)
    cnt |= query(l,mid,o * 2,x,y);
    if (y > mid)
    cnt |= query(mid + 1,r,o * 2 + 1,x,y);
    return cnt;
}

int main()
{
    scanf("%d%d%d",&L,&T,&O);
    build(1,L,1);
    for (int i = 1; i <= O; i++)
    {
        char op[2];
        int a,b,c;
        scanf("%s",op);
        if (op[0] == 'C')
        {
        scanf("%d%d%d",&a,&b,&c);
        if (a > b)
        swap(a,b);
        update(1,L,1,a,b,1 << (c - 1));
        }
        else
        {
            scanf("%d%d",&a,&b);
            if (a > b)
            swap(a,b);
            int ans = query(1,L,1,a,b);
            int cnt = 0;
            while (ans)
            {
                if (ans & 1)
                cnt++;
                ans >>= 1;
            }
            printf("%d\n",cnt);
        }
    }

    return 0;
}