pku2777 Count Color

思路：用位图记录区间已涂上的颜色

 

#include  < iostream >
using   namespace  std;

#define  clr(x) memset(x,0,sizeof(x))
#define  max(a,b) (a>b?a:b)
#define  min(a,b) (a<b?a:b)
#define  MAXN 100002

struct  Node{
     int  l,r;
    __int64 s;
}nod[MAXN * 3 ];


inline  bool  single(__int64 a){
     return  (((a - 2 )  &  a) == 0 );
}


void  init( int  tag, int  l, int  r){
    nod[tag].l = l;
    nod[tag].r = r;
    nod[tag].s = 2 ;
     if (l == r){
         return ;
    }
    init(tag * 2 ,l,(l + r) / 2 );
    init(tag * 2 + 1 ,(l + r) / 2 + 1 ,r);
}


void  query( int  tag, int  l, int  r,__int64  & ss){
     if (l <= nod[tag].l  &&  nod[tag].r <= r){
        ss |= nod[tag].s;
         return ;
    }
     if (single(nod[tag].s)){ // 若该结点所示区间为单色，则其儿子所示区间也为单色，故无需再查看以该结点为根的树
        ss |= nod[tag].s;
         return ;
    }
     if (l <= nod[tag * 2 ].r)
        query(tag * 2 ,l,r,ss);
     if (r >= nod[tag * 2 + 1 ].l)
        query(tag * 2 + 1 ,l,r,ss);
}


void  paint( int  tag, int  l, int  r, int  c){ // 将[l,r]涂成颜色c
     if (l <= nod[tag].l  &&  nod[tag].r <= r){
        nod[tag].s = ( 1 << c);
         return ;
    }
     if (nod[tag].s == ( 1 << c)) // 若该区间已被染成颜色c则无需再染
         return ;
     if (single(nod[tag].s)){ // 若该区间为单色，则在上一次改变中，其儿子没有被改变，又因为需要进入该区间的子树，所以必须先将儿子更新为单色
        nod[tag * 2 ].s = nod[tag].s;
        nod[tag * 2 + 1 ].s = nod[tag].s;
    }
     if (l <= nod[tag * 2 ].r)
        paint(tag * 2 ,l,r,c);
     if (r >= nod[tag * 2 + 1 ].l)
        paint(tag * 2 + 1 ,l,r,c);
    nod[tag].s = nod[tag * 2 ].s | nod[tag * 2 + 1 ].s;
}



int  main(){
     int  L,T,O,i,j,x,y,z,cnt;
     char  ch[ 3 ];
    __int64 ss;
     while (scanf( " %d%d%d " , & L, & T, & O) != EOF){
        init( 1 , 1 ,L);
         for (i = 0 ;i < O;i ++ ){
            scanf( " %s " ,ch);
             if (ch[ 0 ] == ' C ' ){
                scanf( " %d%d%d " , & x, & y, & z);
                 if (x > y){
                     int  t = x;
                    x = y;
                    y = t;
                }
                paint( 1 ,x,y,z);
            }
             else {
                scanf( " %d%d " , & x, & y);
                 if (x > y){
                     int  t = x;
                    x = y;
                    y = t;
                }
                ss = 0 ;
                query( 1 ,x,y,ss);
                cnt = 0 ;
                 for (j = 1 ;j <= T;j ++ ){
                     if (ss & ( 1 << j))
                        cnt ++ ;
                }
                printf( " %d\n " ,cnt);
            }
        }
    }
     return   0 ;
}