POJ Count Color 线段树

题意：

给定一个区域划分成n块，每块对应的序号是1~n

初始情况下所有区域的颜色都是1

有2种操作

P x y c 把x~y区域的颜色变为c

C x y 查询x~y区域的颜色有几种

分析：

我们用线段树处理sum【】存结点对应的区间的颜色如果颜色不唯一就记为-1

ACcode：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <cstring>
using namespace std;
#define maxn 100006
#define mid ((l+r)>>1)
#define tmp (st<<1)
#define lson l,mid,tmp
#define rson mid+1,r,tmp|1
int n,t,q;
int sum[maxn<<2];
int vis[100];
void build(int l,int r,int st){
    sum[st]=1;
    if(l==r)return ;
    build(lson);
    build(rson);
}
void update(int L,int R,int c,int l,int r,int st){
    if(L<=l&&r<=R){
        sum[st]=c;
        return;
    }
    if(sum[st]!=-1)
        sum[tmp|1]=sum[tmp]=sum[st];
    sum[st]=-1;
    if(R<=mid)
        update(L,R,c,lson);
    else if(L>mid)
        update(L,R,c,rson);
    else{
        update(L,mid,c,lson);
        update(mid+1,R,c,rson);
    }
}
void query(int L,int R,int l,int r,int st){
    if(sum[st]!=-1){
        vis[sum[st]]=1;
        return;
    }
    if(R<=mid)query(L,R,lson);
    else if(L>mid)query(L,R,rson);
    else{
        query(L,mid,lson);
        query(mid+1,R,rson);
    }
}
char s[22];
int x,y,temp;
int main(){
    while(scanf("%d%d%d",&n,&t,&q)!=EOF){
        build(1,n,1);
        for(int i=1;i<=q;++i){
            scanf("%s",s);
            if(s[0]=='C'){
                scanf("%d%d%d",&x,&y,&temp);
                if(x>y)x^=y^=x^=y;
                update(x,y,temp,1,n,1);
            }
            else{
                scanf("%d%d",&x,&y);
                if(x>y)x^=y^=x^=y;
                memset(vis,0,sizeof(vis));
                query(x,y,1,n,1);
                int ans=0;
                for(int i=1;i<=t;++i)if(vis[i])ans++;
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}