poj 2777 count color

中文翻译：

题目描述

罗老师有n个点需要染色，一开始这些点染的颜色都是1， 接下来有两种操作：

C a b c: 表示将[a, b]区间的点染成c颜色

P a b: 表示询问[a, b]区间现在有多少种颜色

对于每个询问操作，你需要输出相应的答案。

输入

输入n, C, m，分别表示有n个点，颜色种类最多C种， m个操作

接下来m行，每行输入一种操作

输出

对于每个询问输出相应答案

样例输入

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

样例输出

2
1

提示

【数据规模和约定】

1<=n, m<=100000

1<= C <=30

1<= a, b <=n

1<= c <= C

考虑到C很小，所以可以用状压的思想，直接用一个数来表示有哪些颜色，每次用or运算。

#include<cstdio>
#include<iostream>
#define p1 id<<1
#define p2 id<<1^1
using namespace std;
char s[5];
int n,m,C,x,y,z;
int tree[450000],add[450000];
int a[150000];
void build(int id,int l,int r)
{
    tree[id]=add[id]=1;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(p1,l,mid);
    build(p2,mid+1,r);
}
void update(int id,int l,int r,int x,int y,int z)
{
    if(x<=l&&r<=y)
    {
        tree[id]=(1<<(z-1));
        add[id]=1;
        return;
    }
    if(add[id]==1)         //add[id]=1，表示该区间内颜色统一，=0则有多种颜色
    {
        tree[p1]=tree[id];
        tree[p2]=tree[id];
        add[p1]=1;
        add[p2]=1;
        add[id]=0;
    }
    int mid=(l+r)>>1;
    if(y<=mid) update(p1,l,mid,x,y,z);
    else
    if(x>mid) update(p2,mid+1,r,x,y,z);
    else
    {
        update(p1,l,mid,x,mid,z);
        update(p2,mid+1,r,mid+1,y,z);
    }
    tree[id]=tree[p1]|tree[p2];
}
int query(int id,int l,int r,int x,int y)
{
    if(x<=l&&r<=y) return tree[id];
    if(add[id]==1)
    {
        tree[p1]=tree[id];
        tree[p2]=tree[id];
        add[p1]=1;
        add[p2]=1;
        add[id]=0;
    }
    int mid=(l+r)>>1;
    int res;
    if(y<=mid) res=query(p1,l,mid,x,y);
    else
    if(x>mid) res=query(p2,mid+1,r,x,y);
    else res=query(p1,l,mid,x,mid)|query(p2,mid+1,r,mid+1,y);
    tree[id]=tree[p1]|tree[p2];
    return res;
}
int main()
{
    cin>>n>>C>>m;
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        scanf("%s",s+1);
        if(s[1]=='C')
        {
            scanf("%d%d%d",&x,&y,&z);
            if(x>y) swap(x,y);
            update(1,1,n,x,y,z);
        }
        else
        {
            scanf("%d%d",&x,&y);
            if(x>y) swap(x,y);
            int state=query(1,1,n,x,y);
            int ans=0;
            while(state!=0)
            {
                if(state%2==1) ans++;
                state=state>>1;
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}