Count Color（poj 2777）

题意：

给一个固定长度为L的画板

有两个操作：

C A B C：区间AB内涂上颜色C。

P A B：查询区间AB内颜色种类数。

分析：显然是要用线段树来操作的，设定一个sum[]来维护一个区间内的颜色，若为-1，则说明不是叶子节点，还应在往底下找；

       同时在查询时维护一个vis[]来记录在这个区间内哪个颜色被使用过了。

#include<cstdio>
#include<iostream>
#include<cstring>
#define lson l,m,now*2
#define rson m+1,r,now*2+1
#define M 400010
#define N 35
using namespace std;
int vis[N],sum[M];//当sum[i]=-1是，说明不是底下至少有2种颜色，否则sum[i]是i的颜色 
void build(int l,int r,int now)
{
    sum[now]=1;
    if(l==r)return;
    int m=(l+r)/2;
    build(lson);
    build(rson);
}
void push_down(int now)
{
    sum[now*2]=sum[now];
    sum[now*2+1]=sum[now];
    sum[now]=-1;
}
void change(int x,int y,int v,int l,int r,int now)
{
    if(l>=x&&r<=y)
    {
        sum[now]=v;
        return;
    }
    if(sum[now]==v)return;
    if(sum[now]!=-1)push_down(now);
    int m=(l+r)/2;
    if(x<=m)change(x,y,v,lson);
    if(y>m)change(x,y,v,rson);
}
void query(int x,int y,int l,int r,int now)
{
    if(sum[now]!=-1)
    {
        vis[sum[now]]=1;
        return;
    }
    int m=(l+r)/2;
    if(x<=m)query(x,y,lson);
    if(y>m)query(x,y,rson);
}
int main()
{
    int n,m,p;
    scanf("%d%d%d",&n,&m,&p);
    build(1,n,1);
    for(int i=1;i<=p;i++)
    {
        char c;
        cin>>c;
        if(c=='C')
        {
            int a,b,v;
            scanf("%d%d%d",&a,&b,&v);
            if(a>b)swap(a,b);
            change(a,b,v,1,n,1);
        }
        else
        {
            memset(vis,0,sizeof(vis));
            int a,b,tot=0;
            scanf("%d%d",&a,&b);
            if(a>b)swap(a,b);
            query(a,b,1,n,1);
            for(int i=1;i<=30;i++)
              if(vis[i])tot++;
            printf("%d
",tot);
        }
    }
    return 0;
}