POJ 2777 Count Color (线段树成段更新+二进制思维)

  1 #include <iostream>
  2 #include <cstdio>
  3 #include <cstring>
  4 using namespace std;
  5 const int MAXN = 275000;
  6 struct segtree {
  7     int l , r , val , lazy;
  8 }T[MAXN << 3];
  9 //获取颜色的种类
 10 int get(int n) {
 11     int cont = 0;
 12     while(n) {
 13         if(n & 1) 
 14             cont++;
 15         n >>= 1;
 16     }
 17     return cont;
 18 }
 19 
 20 void init(int p , int l , int r) {
 21     int mid = (l + r) >> 1;
 22     T[p].l = l , T[p].r = r , T[p].lazy = 0;
 23     if(l == r) {
 24         T[p].val = 1;
 25         return ;
 26     }
 27     init(p << 1 , l , mid);
 28     init((p << 1)|1 , mid + 1 , r);
 29     T[p].val = (T[p << 1].val | T[(p << 1)|1].val);
 30 }
 31 
 32 void updata(int p , int l , int r , int val) {
 33     int mid = (T[p].l + T[p].r) >> 1;
 34     if(l == T[p].l && T[p].r == r) {
 35         T[p].val = val;
 36         T[p].lazy = val;
 37         return ;
 38     }
 39     if(T[p].lazy) {
 40         T[p << 1].val = T[(p << 1)|1].val = T[p].lazy;
 41         T[p << 1].lazy = T[(p << 1)|1].lazy = T[p].lazy;
 42         T[p].lazy = 0;
 43     }
 44     if(r <= mid) {
 45         updata(p << 1 , l , r , val);
 46     }
 47     else if(l > mid) {
 48         updata((p << 1)|1 , l , r , val);
 49     }
 50     else {
 51         updata(p << 1 , l , mid , val);
 52         updata((p << 1)|1 , mid + 1 , r , val);
 53     }
 54     T[p].val = (T[p << 1].val | T[(p << 1)|1].val);
 55 }
 56 //返回颜色种类的二进制对应的十进制的值
 57 int query(int p , int l , int r) {
 58     int mid = (T[p].l + T[p].r) >> 1;
 59     if(T[p].l == l && T[p].r == r) {
 60         return T[p].val;
 61     }
 62     if(T[p].lazy) {
 63         T[p << 1].val = T[(p << 1)|1].val = T[p].lazy;
 64         T[p << 1].lazy = T[(p << 1)|1].lazy = T[p].lazy;
 65         T[p].lazy = 0;
 66     }
 67     if(r <= mid) {
 68         return query(p << 1 , l , r);
 69     }
 70     else if(l > mid) {
 71         return query((p << 1)|1 , l , r);
 72     }
 73     else {
 74         return query(p << 1 , l , mid) | query((p << 1)|1 , mid + 1 , r);
 75     }
 76 }
 77 
 78 int main()
 79 {
 80     int L , t , n , l , r , c;
 81     char str[5];
 82     while(~scanf("%d %d %d" , &L , &t , &n)) {
 83         init(1 , 1 , L);
 84         while(n--) {
 85             scanf("%s" , str);
 86             if(str[0] == ‘C‘) {
 87                 scanf("%d %d %d" , &l , &r , &c);
 88                 int temp = l + r;
 89                 l = min(l , r);
 90                 r = temp - l;
 91                 updata(1 , l , r , (1 << (c - 1)));
 92             }
 93             else {
 94                 scanf("%d %d" , &l , &r);
 95                 int temp = l + r;
 96                 l = min(l , r);
 97                 r = temp - l;
 98                 int res = get(query(1 , l , r));
 99                 printf("%d\n" , res);
100             }
101         }
102     }
103 }