洛谷 P1558 色板游戏

洛谷 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

---

题解

线段树+二进制状态压缩

---

代码

#include<cstdio>
#include<cstring>
using namespace std;

int n,m,q,rt,ans;
struct tree
{
    int l,r,sum,tag;
} tr[300005];

int readln()
{
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x;
}

int max(int x,int y){return x>y?x:y;}

void build(int l,int r,int rt)
{
    tr[rt].l=l;tr[rt].r=r;
    tr[rt].sum=1;tr[rt].tag=-1;
    if (l==r) return;
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);build(mid+1,r,rt<<1|1);
}

void pushdown(int rt)
{
    tr[rt<<1].tag=tr[rt].tag;
    tr[rt<<1|1].tag=tr[rt].tag;
    tr[rt<<1].sum=1<<tr[rt].tag;
    tr[rt<<1|1].sum=1<<tr[rt].tag;
    tr[rt].tag=-1;
}

void update(int l,int r,int rt,int x)
{
    int ll=tr[rt].l,rr=tr[rt].r;
    if (ll==l&&rr==r) {tr[rt].tag=x;tr[rt].sum=1<<x;return;}
    if (tr[rt].tag!=-1) pushdown(rt);
    int mid=(ll+rr)>>1;
    if (r<=mid) update(l,r,rt<<1,x);
    else if (mid<l) update(l,r,rt<<1|1,x);
    else {update(l,mid,rt<<1,x);update(mid+1,r,rt<<1|1,x);}
    tr[rt].sum=tr[rt<<1].sum|tr[rt<<1|1].sum;
}

int query(int l,int r,int rt)
{
    int ll=tr[rt].l,rr=tr[rt].r;
    if (ll==l&&rr==r) {return tr[rt].sum;}
    if (tr[rt].tag!=-1) pushdown(rt);
    int mid=(ll+rr)>>1;
    if (r<=mid) return query(l,r,rt<<1);
    else if (mid<l) return query(l,r,rt<<1|1);
    else return query(l,mid,rt<<1)|query(mid+1,r,rt<<1|1);
}

int change(int x)
{
    int cnt=0;
    while (x) {cnt+=x%2;x/=2;}
    return cnt;
}

void sort(int &x,int &y)
{
    int t;
    if (x>y) t=x,x=y,y=t;
}

int main()
{
    n=readln();m=readln();q=readln();
    build(1,n,1);
    while (q--)
    {
        char ch=getchar();while (ch!='C'&&ch!='P') ch=getchar();
        if (ch=='C') {
            int x=readln(),y=readln(),z=readln();
            sort(x,y);
            update(x,y,1,z-1);
        }
        else {
            int x=readln(),y=readln();
            sort(x,y);
            ans=query(x,y,1);
            printf("%d\n",change(ans));
        }
    }
    return 0;
}