POJ 2777 线段树

错误原因：

1. 没有分析清楚什么时候需要向上或者先下传递颜色

2.又把seg[root].l 和 x 搞混淆了

 

思路：采用线段树来处理，节点的成员c用来记录颜色，采用位运算来处理，有30种颜色，二进制位从右往左数第k位表示有无颜色，那么可以用与运算 |  来合并颜色；成员same表示节点所表示区间颜色是否一样，加入same后可以在log(n)时间内实现线段树的insert和query操作。insert时要注意如果区间颜色一样，那么左右子树的颜色也一样，要在insert时传递下去

 

 

 

1. /*
2. PROG:
3. LANG: C++
4. ID: heben991
5. */
6. // algorihm:
7. #include <iostream>
8. #include <algorithm>
9. #include <vector>
10. #include <string>
11. #include <cctype>
12. #include <queue>
13. /*
14. #include <list>
15. #include <stack>
16. #include <set>
17. #include <map>
18. #include <cmath>
19. */
20. using namespace std;
21. #define ma(a,b) (a)>(b)?(a):(b)
22. #define mi(a,b) (a)<(b)?(a):(b)
23. #define FF(i,a,b) for(int i=(a);i<=(b);i++)
24. #define RR(i,b) for(int i=(0);i<(b);i++)
25. #define clr(x) memset(x,0,sizeof(x))
26. #define pb push_back
27. #define mp make_pair
28. #define sz size()
29. #define F  first
30. #define S  second
31. #define vv vector
32. #define ii iterator
33. /*
34. #include <sstream>
35. #include <iterator>
36. #define ssm stringstream
37. */
38. #define cn continue
39. #define br break
40. typedef int type;
41. const int ST = 0;
42. #define out(x) cout << #x << "=" << x << endl
43. template <class T> void show(T a, int n) { for (int i = ST; i < ST+n; i++) { cout<<a[i]<<' '; } cout<<endl; }
44. template <class T> void show(T a, int r, int l) { for (int i = ST; i < ST+r; i++) show(a[i], l); cout<<endl; }
45. const int N=110000,M=100000;
46. struct node
47. {
48.  int l,r,c;
49.  bool same;
50. };
51. node seg[4*N];
52. int l, t, o;
53. void build(int root, int x, int y)
54. {
55.  int len=y-x+1;
56.  int mid=(x+y)/2;
57.  seg[root] = (node) { x,y,1,1};
58.  if(len>1)
59.  {
60.   build(root*2,x,mid);
61.   build(root*2+1,mid+1,y);
62.  }
63. }
64. void insert(int root, int x, int y, int color)
65. {
66.  int mid=(seg[root].l+seg[root].r)/2;
68.  if(x<=seg[root].l && y>=seg[root].r)
69.  {
70.   seg[root].c = 1<<(color-1);
71.   seg[root].same = 1;
72.  }
73.  else
74.  {
75.   if(seg[root].same)
76.   {
77.    seg[root*2].same = seg[root*2+1].same = 1;
78.    seg[root*2].c = seg[root*2+1].c = seg[root].c;
79.   }
80.   if(x<=mid)insert(root*2,x,y,color);
81.   if(y>mid)insert(root*2+1,x,y,color);
82.   seg[root].c = seg[root*2].c | seg[root*2+1].c;
83.   seg[root].same = 0;
84.  }
85. }
86. /*
87. 10 5 8
88. C 1 9 2
89. C 3 8 3
90. P 1 10
91. 2
92. */
93. int query(int root, int x, int y)
94. {
95.  int mid=(seg[root].r+seg[root].l)/2;
97.  if(seg[root].same)return seg[root].c;
98.  if(x<=seg[root].l && y>=seg[root].r)
99.  {
100.   return seg[root].c;
101.  }
102.  {
103.   int ls=0,rs=0;
104.   if(x<=mid)ls = query(root*2,x,y);
105.   if(y>mid)rs = query(root*2+1,x,y);
106.   return ls | rs;
107.  }
108. }
109. int count(int x)
110. {
111.  int ret=0;
112.  //printf("%x ", x);
113.  while(x>0)
114.  {
115.   ret += x & 1;
116.   x>>=1;
117.  }
118.  //printf("%d/n", ret);
119.  return ret;
120. }
121. int main()
122. {
123.  char s[10];
124.  int a, b, c;
125.  scanf("%d%d%d", &l, &t, &o);
126.  //printf("%d %d %d/n", l, t, o);
127.  build(1,1,M);
128.  while(o--)
129.  {
130.   scanf("%s%d%d", s, &a, &b);
131.   //printf("%s %d %d/n", s, a, b);
132.   if(a>b) swap(a,b);
133.   if(s[0]=='C')
134.   {
135.    scanf("%d", &c);
136.    insert(1,a,b,c);
137.   }
138.   else
139.   {
140.    printf("%d/n", count(query(1,a,b)));
141.   }
142.  }
143.     return 0;
144. }