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，这样的话需要你交换A和B)

输出格式

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

输入输出样例

输入 #1

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

输出 #1

2
1

思路：

线段树模板题！！！

代码：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int N=400010;

char u;
int n,t,c,x,y,k;
int sum[N],ad[N];

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

void build(int rt,int l,int r) {
    if (l==r) {
        sum[rt]=1<<1;
        return;
    }
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    pushup(rt);
}

void pushdown(int rt,int l,int r) {
    if (ad[rt]) {
        sum[rt<<1]=(1<<ad[rt]);
        sum[rt<<1|1]=(1<<ad[rt]);
        ad[rt<<1]=ad[rt];
        ad[rt<<1|1]=ad[rt];
        ad[rt]=0;
    }
}

void update(int rt,int l,int r,int x,int y,int k) {
    if (l>y||x>r)
        return;
    if (x<=l&&r<=y) {
        sum[rt]=(1<<k);
        ad[rt]=k;
        return;
    }
    pushdown(rt,l,r);
    int m=(l+r)>>1;
    if (m>=x)
        update(rt<<1,l,m,x,y,k);
    if (m<y)
        update(rt<<1|1,m+1,r,x,y,k);
    pushup(rt);
}

int query(int rt,int l,int r,int x,int y) {
    if(l>y||x>r)
        return 0;
    if(x<=l&&r<=y)
        return sum[rt];
    pushdown(rt,l,r);
    int m=(l+r)>>1,ans=0;
    if(m>=x)
        ans|=query(rt<<1,l,m,x,y);
    if(m<y)
        ans|=query(rt<<1|1,m+1,r,x,y);
    return ans;
}

int main () {
    scanf("%d%d%d",&n,&c,&t);
    build(1,1,n);
    while(t--) {
        cin>>u;
        if(u=='C') {
            scanf("%d%d%d",&x,&y,&k);
            if (x>y)
                swap(x,y);
            update(1,1,n,x,y,k);
        } else {
            scanf("%d%d",&x,&y);
            if(x>y)
                swap(x,y);
            int p=query(1,1,n,x,y),ans=0;
            for (int i=1; i<=c; i++)
                if (p&(1<<i))
                    ans++;
            printf("%d\n",ans);
        }
    }
    return 0;
}