pku 2777 Count Color 解体报告

这到题的题意很明确，给你L长的离散的带子，初始时都图色1。然后给出O个查询和更新条件。对于这样的题一般用线段树比较好做，而且效率很好。由于颜色比较少，可以用二进制的形式来表示（即所有出入的颜色值都要做这样的处理，把它转化为1左移这么多位所对应的值，即1<<(c-1)）,这样在进行颜色合并时就方便多了，直接 A|B。这点搞清楚后，线段树就好写了。
这是我写的代码。

 1 
 2  #include < iostream >
 3  #include < algorithm >
 4  using   namespace  std;
 5  #define  MAXN 100001
 6 
 7  struct  Seg_Tree{
 8    Seg_Tree  * leftptr, * rightptr;
 9     int  left,right;
10     int  col;
11  } * Root;
12 
13  int  L,T,O,cnt;
14  int  nmem;
15  Seg_Tree mem[MAXN * 10 ];
16 
17  Seg_Tree *  CreateNode(){
18    Seg_Tree *  p =& mem[nmem ++ ];
19    memset(p, 0 , sizeof (Seg_Tree));
20     return  p;
21  }
22 
23  Seg_Tree *  CreateTree( int  s, int  e){
24    Seg_Tree *  root = CreateNode();
25    root -> left = s;
26    root -> right = e;
27     if (s != e){
28       int  mid = (s + e) / 2 ;
29      root -> leftptr = CreateTree(s,mid);
30      root -> rightptr = CreateTree(mid + 1 ,e);
31    }
32     return  root;
33  }
34 
35  bool  odd( int  n){ // 判断奇偶的，其实就判断是不是只有一种颜色。
36     return  (n & (n - 1 )) == 0 ;
37  }
38 
39  void  UpdateTree(Seg_Tree *  root, int  s, int  e, int  c){
40       if (s <= root -> left && e >= root -> right){
41        root -> col = c;
42         return ;
43      }
44       if (root -> col == c)  return ;
45       if (odd(root -> col)){
46        root -> leftptr -> col = root -> col;
47        root -> rightptr -> col = root -> col;
48      }
49       int  mid = (root -> left + root -> right) / 2 ;
50       if (s <= mid) UpdateTree(root -> leftptr,s,e,c);
51       if (e > mid) UpdateTree(root -> rightptr,s,e,c);
52      root -> col = root -> leftptr -> col | root -> rightptr -> col;
53  }
54  void  Query(Seg_Tree *  root, int  s, int  e, int   & cnt){
55       if (s <= root -> left && e >= root -> right){
56          cnt = cnt | root -> col;
57           return ;
58      }
59       if (odd(root -> col)){
60          cnt = cnt | root -> col;
61           return ;
62      }
63     int  mid = (root -> left + root -> right) / 2 ;
64     if (s <= mid) Query(root -> leftptr,s,e,cnt);
65     if (e > mid) Query(root -> rightptr,s,e,cnt);
66  }
67  int  cal( int  n){
68     int  cnt = 0 ;
69     while (n > 0 ){
70       if (n % 2 ) cnt ++ ;
71      n >>= 1 ;
72    }
73     return  cnt;
74  }
75  int  main(){
76    nmem = 0 ;
77    scanf( " %d%d%d " , & L, & T, & O);
78    Root = CreateTree( 1 ,L);
79    UpdateTree(Root, 1 ,L, 1 );
80     char  cmd;
81     int  s,e,c;
82     while (O -- ){
83      scanf( "  %c " , & cmd);
84       if (cmd == ' C ' ){
85        scanf( " %d%d%d " , & s, & e, & c);
86         if (s > e) swap(s,e);
87        UpdateTree(Root,s,e, 1 << (c - 1 ));
88      }
89       else {
90        cnt = 0 ;
91        scanf( " %d%d " , & s, & e);
92         if (s > e) swap(s,e);
93        Query(Root,s,e,cnt);
94        printf( " %d\n " ,cal(cnt));
95      }
96    }
97     return   0 ;
98  }
99 

转载于:https://www.cnblogs.com/saintqdd/archive/2007/11/04/948622.html