Interval Tree Topic

今天训练的内容是线段树。

线段树的基本结构，空间大小，常用操作的复杂度已不必多说。主要针对训练中出线的一些问题做一些小的总结

总得来说，对于线段树的编码和调试能力还比较欠缺，这一点希望通过多次的训练来解决

对于各种线段树模型，应用方法则需要经验的积累

 

ZJU 1610：

    线段树的入门题，主要注意的是线段树处理过程中一般如果能覆盖当前区间，操作一般都会终止。这种情况下，如果题意中该区间的操作会影响子区间，那么需要通过一些步骤调整。典型的就是染色。对一个染成同一中颜色的区间的一个子区间染色，我们首先先要将该区间的颜色传递到它的子区间上。见代码line39

1. #include <iostream>
3. using namespace std;
5. const int N = 8010;
7. struct node
8. {
9.     int x, y;
10.     int color;
11. }seg[3*N];
13. int cnt[N];
14. int num[N];
16. void build(int root, int x, int y)
17. {
18.     seg[root] = (node){x, y, -1};
19.     if(y > x + 1)
20.     {
21.         int mid = (x + y) >> 1;
22.         build(root * 2, x, mid);
23.         build(root * 2 + 1, mid, y);
24.     }
25. }
27. void insert(int root, int x, int y, int c)
28. {
29.     if(x <= seg[root].x && y >= seg[root].y)
30.     {
31.         int j = seg[root].color;
32.         seg[root].color = c;
33.     }
34.     else
35.     {
36.         int mid = (seg[root].x + seg[root].y) >> 1;
37.         if(seg[root].color >= 0)
38.         {
39.             seg[root*2].color = seg[root*2 + 1].color = seg[root].color;
40.             seg[root].color = -1;
41.         }
42.         if(x < mid) insert(root * 2, x, y, c);
43.         if(y > mid) insert(root * 2 + 1, x, y, c);
44.     }
45. }
46. void query(int x)
47. {
48.     int i, j;
49.     
50.     if(seg[x].color >= 0)
51.         for(int i = seg[x].x + 1; i <= seg[x].y; i++) cnt[i] = seg[x].color;
52.     else if(seg[x].x + 1 == seg[x].y) cnt[seg[x].y] = -1;
53.     else
54.     {
55.         query(2*x);
56.         query(2*x+1);
57.     }
58. }
59.         
60.         
61. int main()
62. {
63.     int i, j, k, m;
64.     
65.     while(scanf("%d",&m) == 1)
66.     {
67.         memset(cnt, 0, sizeof(cnt));  
68.         memset(num, 0, sizeof(num));
69.               
70.         build(1, 0, 8000);
71.         for(; m; m--)
72.         {
73.             scanf("%d%d%d",&i, &j, &k);
74.             insert(1, i, j, k);
75.         }
76.         query(1);
77.         
78.         i = 1;
79.         while(i <= 8000)
80.         {
81.             while(i <= 8000 && cnt[i]==-1) i++;
82.             if(i > 8000) break;
83.             j = i + 1;
84.             while(j <= 8000 && cnt[j]==cnt[i]) j++;
85.             num[cnt[i]]++;
86.             i = j;
87.         }
88.         for(i = 0; i <= 8000; i++)
89.             if(num[i] != 0) printf("%d %d/n",i, num[i]);
90.         
91.         printf("/n");
92.     }
93.     return 0;
94. }
95.             
96.         
97.     
99.     

PKU 2777：

      和上题相似。注意的地方在于。查询的时候，递归结束既可能是因为覆盖了整个区间，也可能是整个区间只有一中颜色。

line 54

1. #include <iostream>
3. const int N = 100010;
5. struct node
6. {
7.     int x, y;
8.     int color;
9.     bool oneColor;
10. }seg[N<<2];
12. int T;
14. void build(int root, int x, int y)
15. {
16.     seg[root] = (node){x, y, 1, true};
17.     if(x != y)
18.     {
19.         int mid = (x + y) >> 1;
20.         build(root * 2, x, mid);
21.         build(root * 2 + 1, mid + 1, y);
22.     }
23. }
25. void insert(int root, int x, int y, int c)
26. {
27.     if(x <= seg[root].x && y >= seg[root].y){
28.         seg[root].color = (1<<c);
29.         seg[root].oneColor = true;
30.     }
31.     else
32.     {
33.         if(seg[root].oneColor) {
34.             seg[root*2].color = seg[root*2+1].color = seg[root].color;
35.             seg[root*2].oneColor = seg[root*2+1].oneColor = true;
36.             if((1<<c) != seg[root].color) seg[root].oneColor = false;
37.         }
38.         
39.         int mid = (seg[root].x + seg[root].y) >> 1;
40.         if(x <= mid)
41.             insert(root*2, x, y, c);
42.         if(y > mid)
43.             insert(root*2 + 1, x, y, c);
44.         
45.         seg[root].color = seg[root*2].color | seg[root*2+1].color;
46.     }
47. }
49. int query(int root, int x, int y)
50. {
51.     if(x <= seg[root].x && y >= seg[root].y)
52.         return seg[root].color;
53.     
54.     if(seg[root].oneColor) return seg[root].color;
55.     
56.     else
57.     {
58.         int mid = (seg[root].x + seg[root].y) >> 1;
59.         
60.         int ans = 0;
61.         if(x <= mid) ans |= query(root*2, x, y);
62.         if(y > mid)  ans |= query(root*2 + 1, x, y);
63.         
64.         return ans;
65.     }
66. }
68. inline int getNum(int x)
69. {
70.     int i = 0;
71.     while(x != 0) { i += x%2; x/=2;}
72.     return i;
73. }
75. int main()
76. {
77.     int n, m;
78.     int i, j, k;
79.     
80.     scanf("%d%d%d",&n, &T, &m);
81.     build(1, 1, n);
82.     
83.     for(; m; m--)
84.     {
85.         getchar();
86.         char ch = getchar();
87.         if(ch == 'C')
88.         { 
89.             scanf("%d%d%d",&i, &j, &k);
90.             if(i > j) { int t = i; i = j; j = t;}
91.             insert(1, i, j, k-1);
92.         }
93.         else
94.         {
95.             scanf("%d%d",&i, &j);
96.             if(i > j) { int t = i; i = j; j = t;}
97.             printf("%d/n",getNum(query(1, i, j)));
98.         }
99.     }
100.     return 0;
101. }
102.