线段树详解

转载自:https://blog.csdn.net/metalseed/article/details/8039326


一:线段树基本概念

1:概述

线段树,类似区间树,是一个完全二叉树,它在各个节点保存一条线段(数组中的一段子数组),主要用于高效解决连续区间的动态查询问题,由于二叉结构的特性,它基本能保持每个操作的复杂度为O(lgN)!

性质:父亲的区间是[a,b],(c=(a+b)/2)左儿子的区间是[a,c],右儿子的区间是[c+1,b],线段树需要的空间为数组大小的四倍


2:基本操作(demo用的是查询区间最小值)

线段树的主要操作有:

(1):线段树的构造 void build(int node, int begin, int end);

主要思想是递归构造,如果当前节点记录的区间只有一个值,则直接赋值,否则递归构造左右子树,最后回溯的时候给当前节点赋值


 
 
  1. #include <iostream>
  2. using namespace std;
  3. const int maxind = 256;
  4. int segTree[maxind * 4 + 10];
  5. int array[maxind];
  6. /* 构造函数,得到线段树 */
  7. void build(int node, int begin, int end)
  8. {
  9. if (begin == end)
  10. segTree[node] = array[begin]; /* 只有一个元素,节点记录该单元素 */
  11. else
  12. {
  13. /* 递归构造左右子树 */
  14. build( 2*node, begin, (begin+end)/ 2);
  15. build( 2*node+ 1, (begin+end)/ 2+ 1, end);
  16. /* 回溯时得到当前node节点的线段信息 */
  17. if (segTree[ 2 * node] <= segTree[ 2 * node + 1])
  18. segTree[node] = segTree[ 2 * node];
  19. else
  20. segTree[node] = segTree[ 2 * node + 1];
  21. }
  22. }
  23. int main()
  24. {
  25. array[ 0] = 1, array[ 1] = 2, array[ 2] = 2, array[ 3] = 4, array[ 4] = 1, array[ 5] = 3;
  26. build( 1, 0, 5);
  27. for( int i = 1; i<= 20; ++i)
  28. cout<< "seg"<< i << "=" <<segTree[i] << endl;
  29. return 0;
  30. }
 此build构造成的树如图:

(2):区间查询int query(int node, int begin, int end, int left, int right);

(其中node为当前查询节点,begin,end为当前节点存储的区间,left,right为此次query所要查询的区间)

主要思想是把所要查询的区间[a,b]划分为线段树上的节点,然后将这些节点代表的区间合并起来得到所需信息

比如前面一个图中所示的树,如果询问区间是[0,2],或者询问的区间是[3,3],不难直接找到对应的节点回答这一问题。但并不是所有的提问都这么容易回答,比如[0,3],就没有哪一个节点记录了这个区间的最小值。当然,解决方法也不难找到:把[0,2][3,3]两个区间(它们在整数意义上是相连的两个区间)的最小值合并起来,也就是求这两个最小值的最小值,就能求出[0,3]范围的最小值。同理,对于其他询问的区间,也都可以找到若干个相连的区间,合并后可以得到询问的区间。


 
 
  1. int query(int node, int begin, int end, int left, int right)
  2. {
  3. int p1, p2;
  4. /* 查询区间和要求的区间没有交集 */
  5. if (left > end || right < begin)
  6. return -1;
  7. /* if the current interval is included in */
  8. /* the query interval return segTree[node] */
  9. if (begin >= left && end <= right)
  10. return segTree[node];
  11. /* compute the minimum position in the */
  12. /* left and right part of the interval */
  13. p1 = query( 2 * node, begin, (begin + end) / 2, left, right);
  14. p2 = query( 2 * node + 1, (begin + end) / 2 + 1, end, left, right);
  15. /* return the expect value */
  16. if (p1 == -1)
  17. return p2;
  18. if (p2 == -1)
  19. return p1;
  20. if (p1 <= p2)
  21. return p1;
  22. return p2;
  23. }

可见,这样的过程一定选出了尽量少的区间,它们相连后正好涵盖了整个[left,right],没有重复也没有遗漏。同时,考虑到线段树上每层的节点最多会被选取2个,一共选取的节点数也是O(log n)的,因此查询的时间复杂度也是O(log n)。

线段树并不适合所有区间查询情况,它的使用条件是“相邻的区间的信息可以被合并成两个区间的并区间的信息”。即问题是可以被分解解决的。



(3):区间或节点的更新 及 线段树的动态维护update (这是线段树核心价值所在,节点中的标记域可以解决N多种问题)

动态维护需要用到标记域,延迟标记等。

a:单节点更新


 
 
  1. void Updata(int node, int begin, int end, int ind, int add)/*单节点更新*/
  2. {
  3. if( begin == end )
  4. {
  5. segTree[node] += add;
  6. return ;
  7. }
  8. int m = ( left + right ) >> 1;
  9. if(ind <= m)
  10. Updata(node * 2,left, m, ind, add);
  11. else
  12. Updata(node * 2 + 1, m + 1, right, ind, add);
  13. /*回溯更新父节点*/
  14. segTree[node] = min(segTree[node * 2], segTree[node * 2 + 1]);
  15. }

b:区间更新(线段树中最有用的)

需要用到延迟标记,每个结点新增加一个标记,记录这个结点是否被进行了某种修改操作(这种修改操作会影响其子结点)。对于任意区间的修改,我们先按照查询的方式将其划分成线段树中的结点,然后修改这些结点的信息,并给这些结点标上代表这种修改操作的标记。在修改和查询的时候,如果我们到了一个结点p,并且决定考虑其子结点,那么我们就要看看结点p有没有标记,如果有,就要按照标记修改其子结点的信息,并且给子结点都标上相同的标记,同时消掉p的标记。(优点在于,不用将区间内的所有值都暴力更新,大大提高效率,因此区间更新是最优用的操作)

void Change来自dongxicheng.org


 
 
  1. void Change(node *p, int a, int b) /* 当前考察结点为p,修改区间为(a,b]*/
  2. {
  3. if (a <= p->Left && p->Right <= b)
  4. /* 如果当前结点的区间包含在修改区间内*/
  5. {
  6. ...... /* 修改当前结点的信息,并标上标记*/
  7. return;
  8. }
  9. Push_Down(p); /* 把当前结点的标记向下传递*/
  10. int mid = (p->Left + p->Right) / 2; /* 计算左右子结点的分隔点
  11. if (a < mid) Change(p->Lch, a, b); /* 和左孩子有交集,考察左子结点*/
  12. if (b > mid) Change(p->Rch, a, b); /* 和右孩子有交集,考察右子结点*/
  13. Update(p); /* 维护当前结点的信息(因为其子结点的信息可能有更改)*/
  14. }



3:主要应用

(1):区间最值查询问题 (见模板1)

(2):连续区间修改或者单节点更新的动态查询问题 (见模板2)

(3):多维空间的动态查询 (见模板3)


二:典型模板

模板1:

RMQ,查询区间最值下标---min


 
 
  1. #include<iostream>
  2. using namespace std;
  3. #define MAXN 100
  4. #define MAXIND 256 //线段树节点个数
  5. //构建线段树,目的:得到M数组.
  6. void build(int node, int b, int e, int M[], int A[])
  7. {
  8. if (b == e)
  9. M[node] = b; //只有一个元素,只有一个下标
  10. else
  11. {
  12. build( 2 * node, b, (b + e) / 2, M, A);
  13. build( 2 * node + 1, (b + e) / 2 + 1, e, M, A);
  14. if (A[M[ 2 * node]] <= A[M[ 2 * node + 1]])
  15. M[node] = M[ 2 * node];
  16. else
  17. M[node] = M[ 2 * node + 1];
  18. }
  19. }
  20. //找出区间 [i, j] 上的最小值的索引
  21. int query(int node, int b, int e, int M[], int A[], int i, int j)
  22. {
  23. int p1, p2;
  24. //查询区间和要求的区间没有交集
  25. if (i > e || j < b)
  26. return -1;
  27. if (b >= i && e <= j)
  28. return M[node];
  29. p1 = query( 2 * node, b, (b + e) / 2, M, A, i, j);
  30. p2 = query( 2 * node + 1, (b + e) / 2 + 1, e, M, A, i, j);
  31. //return the position where the overall
  32. //minimum is
  33. if (p1 == -1)
  34. return M[node] = p2;
  35. if (p2 == -1)
  36. return M[node] = p1;
  37. if (A[p1] <= A[p2])
  38. return M[node] = p1;
  39. return M[node] = p2;
  40. }
  41. int main()
  42. {
  43. int M[MAXIND]; //下标1起才有意义,否则不是二叉树,保存下标编号节点对应区间最小值的下标.
  44. memset(M, -1, sizeof(M));
  45. int a[]={ 3, 4, 5, 7, 2, 1, 0, 3, 4, 5};
  46. build( 1, 0, sizeof(a)/ sizeof(a[ 0]) -1, M, a);
  47. cout<<query( 1, 0, sizeof(a)/ sizeof(a[ 0]) -1, M, a, 0, 5)<< endl;
  48. return 0;
  49. }



模板2:

连续区间修改或者单节点更新的动态查询问题 (此模板查询区间和)


 
 
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. #define root 1 , N , 1
  7. #define LL long long
  8. const int maxn = 111111;
  9. LL add[maxn<< 2];
  10. LL sum[maxn<< 2];
  11. void PushUp(int rt) {
  12. sum[rt] = sum[rt<< 1] + sum[rt<< 1| 1];
  13. }
  14. void PushDown(int rt,int m) {
  15. if (add[rt]) {
  16. add[rt<< 1] += add[rt];
  17. add[rt<< 1| 1] += add[rt];
  18. sum[rt<< 1] += add[rt] * (m - (m >> 1));
  19. sum[rt<< 1| 1] += add[rt] * (m >> 1);
  20. add[rt] = 0;
  21. }
  22. }
  23. void build(int l,int r,int rt) {
  24. add[rt] = 0;
  25. if (l == r) {
  26. scanf( "%lld",&sum[rt]);
  27. return ;
  28. }
  29. int m = (l + r) >> 1;
  30. build(lson);
  31. build(rson);
  32. PushUp(rt);
  33. }
  34. void update(int L,int R,int c,int l,int r,int rt) {
  35. if (L <= l && r <= R) {
  36. add[rt] += c;
  37. sum[rt] += (LL)c * (r - l + 1);
  38. return ;
  39. }
  40. PushDown(rt , r - l + 1);
  41. int m = (l + r) >> 1;
  42. if (L <= m) update(L , R , c , lson);
  43. if (m < R) update(L , R , c , rson);
  44. PushUp(rt);
  45. }
  46. LL query(int L,int R,int l,int r,int rt) {
  47. if (L <= l && r <= R) {
  48. return sum[rt];
  49. }
  50. PushDown(rt , r - l + 1);
  51. int m = (l + r) >> 1;
  52. LL ret = 0;
  53. if (L <= m) ret += query(L , R , lson);
  54. if (m < R) ret += query(L , R , rson);
  55. return ret;
  56. }
  57. int main() {
  58. int N , Q;
  59. scanf( "%d%d",&N,&Q);
  60. build(root);
  61. while (Q --) {
  62. char op[ 2];
  63. int a , b , c;
  64. scanf( "%s",op);
  65. if (op[ 0] == 'Q') {
  66. scanf( "%d%d",&a,&b);
  67. printf( "%lld\n",query(a , b ,root));
  68. } else {
  69. scanf( "%d%d%d",&a,&b,&c);
  70. update(a , b , c , root);
  71. }
  72. }
  73. return 0;
  74. }


模板3:

多维空间的动态查询



三:练习题目

下面是hh线段树代码,典型练习哇~

在代码前先介绍一些我的线段树风格:

  • maxn是题目给的最大区间,而节点数要开4倍,确切的来说节点数要开大于maxn的最小2x的两倍
  • lson和rson分辨表示结点的左儿子和右儿子,由于每次传参数的时候都固定是这几个变量,所以可以用预定于比较方便的表示
  • 以前的写法是另外开两个个数组记录每个结点所表示的区间,其实这个区间不必保存,一边算一边传下去就行,只需要写函数的时候多两个参数,结合lson和rson的预定义可以很方便
  • PushUP(int rt)是把当前结点的信息更新到父结点
  • PushDown(int rt)是把当前结点的信息更新给儿子结点
  • rt表示当前子树的根(root),也就是当前所在的结点

整理这些题目后我觉得线段树的题目整体上可以分成以下四个部分:



单点更新:最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来


code:


 
 
  1. #include<cstring>
  2. #include<iostream>
  3. #define M 50005
  4. #define lson l,m,rt<<1
  5. #define rson m+1,r,rt<<1|1
  6. /*left,right,root,middle*/
  7. int sum[M<< 2];
  8. inline void PushPlus(int rt)
  9. {
  10. sum[rt] = sum[rt<< 1] + sum[rt<< 1| 1];
  11. }
  12. void Build(int l, int r, int rt)
  13. {
  14. if(l == r)
  15. {
  16. scanf( "%d", &sum[rt]);
  17. return ;
  18. }
  19. int m = ( l + r )>> 1;
  20. Build(lson);
  21. Build(rson);
  22. PushPlus(rt);
  23. }
  24. void Updata(int p, int add, int l, int r, int rt)
  25. {
  26. if( l == r )
  27. {
  28. sum[rt] += add;
  29. return ;
  30. }
  31. int m = ( l + r ) >> 1;
  32. if(p <= m)
  33. Updata(p, add, lson);
  34. else
  35. Updata(p, add, rson);
  36. PushPlus(rt);
  37. }
  38. int Query(int L,int R,int l,int r,int rt)
  39. {
  40. if( L <= l && r <= R )
  41. {
  42. return sum[rt];
  43. }
  44. int m = ( l + r ) >> 1;
  45. int ans= 0;
  46. if(L<=m )
  47. ans+=Query(L,R,lson);
  48. if(R>m)
  49. ans+=Query(L,R,rson);
  50. return ans;
  51. }
  52. int main()
  53. {
  54. int T, n, a, b;
  55. scanf( "%d",&T);
  56. for( int i = 1; i <= T; ++i )
  57. {
  58. printf( "Case %d:\n",i);
  59. scanf( "%d",&n);
  60. Build( 1,n, 1);
  61. char op[ 10];
  62. while( scanf( "%s",op) &&op[ 0]!= 'E' )
  63. {
  64. scanf( "%d %d", &a, &b);
  65. if(op[ 0] == 'Q')
  66. printf( "%d\n",Query(a,b, 1,n, 1));
  67. else if(op[ 0] == 'S')
  68. Updata(a,-b, 1,n, 1);
  69. else
  70. Updata(a,b, 1,n, 1);
  71. }
  72. }
  73. return 0;
  74. }

hdu1754 I Hate It
题意:O(-1)
思路:O(-1)
线段树功能:update:单点替换 query:区间最值


  
  
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. const int maxn = 222222;
  7. int MAX[maxn<< 2];
  8. void PushUP(int rt) {
  9. MAX[rt] = max(MAX[rt<< 1] , MAX[rt<< 1| 1]);
  10. }
  11. void build(int l,int r,int rt) {
  12. if (l == r) {
  13. scanf( "%d",&MAX[rt]);
  14. return ;
  15. }
  16. int m = (l + r) >> 1;
  17. build(lson);
  18. build(rson);
  19. PushUP(rt);
  20. }
  21. void update(int p,int sc,int l,int r,int rt) {
  22. if (l == r) {
  23. MAX[rt] = sc;
  24. return ;
  25. }
  26. int m = (l + r) >> 1;
  27. if (p <= m) update(p , sc , lson);
  28. else update(p , sc , rson);
  29. PushUP(rt);
  30. }
  31. int query(int L,int R,int l,int r,int rt) {
  32. if (L <= l && r <= R) {
  33. return MAX[rt];
  34. }
  35. int m = (l + r) >> 1;
  36. int ret = 0;
  37. if (L <= m) ret = max(ret , query(L , R , lson));
  38. if (R > m) ret = max(ret , query(L , R , rson));
  39. return ret;
  40. }
  41. int main() {
  42. int n , m;
  43. while (~ scanf( "%d%d",&n,&m)) {
  44. build( 1 , n , 1);
  45. while (m --) {
  46. char op[ 2];
  47. int a , b;
  48. scanf( "%s%d%d",op,&a,&b);
  49. if (op[ 0] == 'Q') printf( "%d\n",query(a , b , 1 , n , 1));
  50. else update(a , b , 1 , n , 1);
  51. }
  52. }
  53. return 0;
  54. }

hdu1394 Minimum Inversion Number
题意:求Inversion后的最小逆序数
思路:用O(nlogn)复杂度求出最初逆序数后,就可以用O(1)的复杂度分别递推出其他解
线段树功能:update:单点增减 query:区间求和


  
  
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. const int maxn = 5555;
  7. int sum[maxn<< 2];
  8. void PushUP(int rt) {
  9. sum[rt] = sum[rt<< 1] + sum[rt<< 1| 1];
  10. }
  11. void build(int l,int r,int rt) {
  12. sum[rt] = 0;
  13. if (l == r) return ;
  14. int m = (l + r) >> 1;
  15. build(lson);
  16. build(rson);
  17. }
  18. void update(int p,int l,int r,int rt) {
  19. if (l == r) {
  20. sum[rt] ++;
  21. return ;
  22. }
  23. int m = (l + r) >> 1;
  24. if (p <= m) update(p , lson);
  25. else update(p , rson);
  26. PushUP(rt);
  27. }
  28. int query(int L,int R,int l,int r,int rt) {
  29. if (L <= l && r <= R) {
  30. return sum[rt];
  31. }
  32. int m = (l + r) >> 1;
  33. int ret = 0;
  34. if (L <= m) ret += query(L , R , lson);
  35. if (R > m) ret += query(L , R , rson);
  36. return ret;
  37. }
  38. int x[maxn];
  39. int main() {
  40. int n;
  41. while (~ scanf( "%d",&n)) {
  42. build( 0 , n - 1 , 1);
  43. int sum = 0;
  44. for ( int i = 0 ; i < n ; i ++) {
  45. scanf( "%d",&x[i]);
  46. sum += query(x[i] , n - 1 , 0 , n - 1 , 1);
  47. update(x[i] , 0 , n - 1 , 1);
  48. }
  49. int ret = sum;
  50. for ( int i = 0 ; i < n ; i ++) {
  51. sum += n - x[i] - x[i] - 1;
  52. ret = min(ret , sum);
  53. }
  54. printf( "%d\n",ret);
  55. }
  56. return 0;
  57. }

hdu2795 Billboard
题意:h*w的木板,放进一些1*L的物品,求每次放空间能容纳且最上边的位子
思路:每次找到最大值的位子,然后减去L
线段树功能:query:区间求最大值的位子(直接把update的操作在query里做了)


  
  
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. const int maxn = 222222;
  7. int h , w , n;
  8. int MAX[maxn<< 2];
  9. void PushUP(int rt) {
  10. MAX[rt] = max(MAX[rt<< 1] , MAX[rt<< 1| 1]);
  11. }
  12. void build(int l,int r,int rt) {
  13. MAX[rt] = w;
  14. if (l == r) return ;
  15. int m = (l + r) >> 1;
  16. build(lson);
  17. build(rson);
  18. }
  19. int query(int x,int l,int r,int rt) {
  20. if (l == r) {
  21. MAX[rt] -= x;
  22. return l;
  23. }
  24. int m = (l + r) >> 1;
  25. int ret = (MAX[rt<< 1] >= x) ? query(x , lson) : query(x , rson);
  26. PushUP(rt);
  27. return ret;
  28. }
  29. int main() {
  30. while (~ scanf( "%d%d%d",&h,&w,&n)) {
  31. if (h > n) h = n;
  32. build( 1 , h , 1);
  33. while (n --) {
  34. int x;
  35. scanf( "%d",&x);
  36. if (MAX[ 1] < x) puts( "-1");
  37. else printf( "%d\n",query(x , 1 , h , 1));
  38. }
  39. }
  40. return 0;
  41. }

成段更新(通常这对初学者来说是一道坎),需要用到延迟标记(或者说懒惰标记),简单来说就是每次更新的时候不要更新到底,用延迟标记使得更新延迟到下次需要更新or询问到的时候

hdu1698 Just a Hook
题意:O(-1)
思路:O(-1)
线段树功能:update:成段替换 (由于只query一次总区间,所以可以直接输出1结点的信息)

  
  
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. const int maxn = 111111;
  7. int h , w , n;
  8. int col[maxn<< 2];
  9. int sum[maxn<< 2];
  10. void PushUp(int rt) {
  11. sum[rt] = sum[rt<< 1] + sum[rt<< 1| 1];
  12. }
  13. void PushDown(int rt,int m) {
  14. if (col[rt]) {
  15. col[rt<< 1] = col[rt<< 1| 1] = col[rt];
  16. sum[rt<< 1] = (m - (m >> 1)) * col[rt];
  17. sum[rt<< 1| 1] = (m >> 1) * col[rt];
  18. col[rt] = 0;
  19. }
  20. }
  21. void build(int l,int r,int rt) {
  22. col[rt] = 0;
  23. sum[rt] = 1;
  24. if (l == r) return ;
  25. int m = (l + r) >> 1;
  26. build(lson);
  27. build(rson);
  28. PushUp(rt);
  29. }
  30. void update(int L,int R,int c,int l,int r,int rt) {
  31. if (L <= l && r <= R) {
  32. col[rt] = c;
  33. sum[rt] = c * (r - l + 1);
  34. return ;
  35. }
  36. PushDown(rt , r - l + 1);
  37. int m = (l + r) >> 1;
  38. if (L <= m) update(L , R , c , lson);
  39. if (R > m) update(L , R , c , rson);
  40. PushUp(rt);
  41. }
  42. int main() {
  43. int T , n , m;
  44. scanf( "%d",&T);
  45. for ( int cas = 1 ; cas <= T ; cas ++) {
  46. scanf( "%d%d",&n,&m);
  47. build( 1 , n , 1);
  48. while (m --) {
  49. int a , b , c;
  50. scanf( "%d%d%d",&a,&b,&c);
  51. update(a , b , c , 1 , n , 1);
  52. }
  53. printf( "Case %d: The total value of the hook is %d.\n",cas , sum[ 1]);
  54. }
  55. return 0;
  56. }

poj3468 A Simple Problem with Integers
题意:O(-1)
思路:O(-1)
线段树功能:update:成段增减 query:区间求和


  
  
  1. #include <cstdio>
  2. #include <algorithm>
  3. using namespace std;
  4. #define lson l , m , rt << 1
  5. #define rson m + 1 , r , rt << 1 | 1
  6. #define LL long long
  7. const int maxn = 111111;
  8. LL add[maxn<< 2];
  9. LL sum[maxn<< 2];
  10. void PushUp(int rt) {
  11. sum[rt] = sum[rt<< 1] + sum[rt<< 1| 1];
  12. }
  13. void PushDown(int rt,int m) {
  14. if (add[rt]) {
  15. add[rt<< 1] += add[rt];
  16. add[rt<< 1| 1] += add[rt];
  17. sum[rt<< 1] += add[rt] * (m - (m >> 1));
  18. sum[rt<< 1| 1] += add[rt] * (m >> 1);
  19. add[rt] = 0;
  20. }
  21. }
  22. void build(int l,int r,int rt) {
  23. add[rt] = 0;
  24. if (l == r) {
  25. scanf( "%lld",&sum[rt]);
  26. return ;
  27. }
  28. int m = (l + r) >> 1;
  29. build(lson);
  30. build(rson);
  31. PushUp(rt);
  32. }
  33. void update(int L,int R,int c,int l,int r,int rt) {
  34. if (L <= l && r <= R) {
  35. add[rt] += c;
  36. sum[rt] += (LL)c * (r - l + 1);
  37. return ;
  38. }
  39. PushDown(rt , r - l + 1);
  40. int m = (l + r) >> 1;
  41. if (L <= m) update(L , R , c , lson);
  42. if (m < R) update(L , R , c , rson);
  43. PushUp(rt);
  44. }
  45. LL query(int L,int R,int l,int r,int rt) {
  46. if (L <= l && r <= R) {
  47. return sum[rt];
  48. }
  49. PushDown(rt , r - l + 1);
  50. int m = (l + r) >> 1;
  51. LL ret = 0;
  52. if (L <= m) ret += query(L , R , lson);
  53. if (m < R) ret += query(L , R , rson);
  54. return ret;
  55. }
  56. int main() {
  57. int N , Q;
  58. scanf( "%d%d",&N,&Q);
  59. build( 1 , N , 1);
  60. while (Q --) {
  61. char op[ 2];
  62. int a , b , c;
  63. scanf( "%s",op);
  64. if (op[ 0] == 'Q') {
  65. scanf( "%d%d",&a,&b);
  66. printf( "%lld\n",query(a , b , 1 , N , 1));
  67. } else {
  68. scanf( "%d%d%d",&a,&b,&c);
  69. update(a , b , c , 1 , N , 1);
  70. }
  71. }
  72. return 0;
  73. }

poj2528 Mayor’s posters
题意:在墙上贴海报,海报可以互相覆盖,问最后可以看见几张海报
思路:这题数据范围很大,直接搞超时+超内存,需要离散化:
离散化简单的来说就是只取我们需要的值来用,比如说区间[1000,2000],[1990,2012] 我们用不到[-∞,999][1001,1989][1991,1999][2001,2011][2013,+∞]这些值,所以我只需要1000,1990,2000,2012就够了,将其分别映射到0,1,2,3,在于复杂度就大大的降下来了
所以离散化要保存所有需要用到的值,排序后,分别映射到1~n,这样复杂度就会小很多很多
而这题的难点在于每个数字其实表示的是一个单位长度(并非一个点),这样普通的离散化会造成许多错误(包括我以前的代码,poj这题数据奇弱)
给出下面两个简单的例子应该能体现普通离散化的缺陷:
例子一:1-10 1-4 5-10
例子二:1-10 1-4 6-10
普通离散化后都变成了[1,4][1,2][3,4]
线段2覆盖了[1,2],线段3覆盖了[3,4],那么线段1是否被完全覆盖掉了呢?
例子一是完全被覆盖掉了,而例子二没有被覆盖

为了解决这种缺陷,我们可以在排序后的数组上加些处理,比如说[1,2,6,10]
如果相邻数字间距大于1的话,在其中加上任意一个数字,比如加成[1,2,3,6,7,10],然后再做线段树就好了.
线段树功能:update:成段替换 query:简单hash


  
  
  1. #include <cstdio>
  2. #include <cstring>
  3. #include <algorithm>
  4. using namespace std;
  5. #define lson l , m , rt << 1
  6. #define rson m + 1 , r , rt << 1 | 1
  7. const int maxn = 11111;
  8. bool hash[maxn];
  9. int li[maxn] , ri[maxn];
  10. int X[maxn* 3];
  11. int col[maxn<< 4];
  12. int cnt;
  13. void PushDown(int rt) {
  14. if (col[rt] != -1) {
  15. col[rt<< 1] = col[rt<< 1| 1] = col[rt];
  16. col[rt] = -1;
  17. }
  18. }
  19. void update(int L,int R,int c,int l,int r,int rt) {
  20. if (L <= l && r <= R) {
  21. col[rt] = c;
  22. return ;
  23. }
  24. PushDown(rt);
  25. int m = (l + r) >> 1;
  26. if (L <= m) update(L , R , c , lson);
  27. if (m < R) update(L , R , c , rson);
  28. }
  29. void query(int l,int r,int rt) {
  30. if (col[rt] != -1) {
  31. if (!hash[col[rt]]) cnt ++;
  32. hash[ col[rt] ] = true;
  33. return ;
  34. }
  35. if (l == r) return ;
  36. int m = (l + r) >> 1;
  37. query(lson);
  38. query(rson);
  39. }
  40. int Bin(int key,int n,int X[]) {
  41. int l = 0 , r = n - 1;
  42. while (l <= r) {
  43. int m = (l + r) >> 1;
  44. if (X[m] == key) return m;
  45. if (X[m] < key) l = m + 1;
  46. else r = m - 1;
  47. }
  48. return -1;
  49. }
  50. int main() {
  51. int T , n;
  52. scanf( "%d",&T);
  53. while (T --) {
  54. scanf( "%d",&n);
  55. int nn = 0;
  56. for ( int i = 0 ; i < n ; i ++) {
  57. scanf( "%d%d",&li[i] , &ri[i]);
  58. X[nn++] = li[i];
  59. X[nn++] = ri[i];
  60. }
  61. sort(X , X + nn);
  62. int m = 1;
  63. for ( int i = 1 ; i < nn; i ++) {
  64. if (X[i] != X[i -1]) X[m ++] = X[i];
  65. }
  66. for ( int i = m - 1 ; i > 0 ; i --) {
  67. if (X[i] != X[i -1] + 1) X[m ++] = X[i -1] + 1;
  68. }
  69. sort(X , X + m);
  70. memset(col , -1 , sizeof(col));
  71. for ( int i = 0 ; i < n ; i ++) {
  72. int l = Bin(li[i] , m , X);
  73. int r = Bin(ri[i] , m , X);
  74. update(l , r , i , 0 , m , 1);
  75. }
  76. cnt = 0;
  77. memset(hash , false , sizeof(hash));
  78. query( 0 , m , 1);
  79. printf( "%d\n",cnt);
  80. }
  81. return 0;
  82. }

poj3225 Help with Intervals
题意:区间操作,交,并,补等
思路:
我们一个一个操作来分析:(用0和1表示是否包含区间,-1表示该区间内既有包含又有不包含)
U:把区间[l,r]覆盖成1
I:把[-∞,l)(r,∞]覆盖成0
D:把区间[l,r]覆盖成0
C:把[-∞,l)(r,∞]覆盖成0 , 且[l,r]区间0/1互换
S:[l,r]区间0/1互换

成段覆盖的操作很简单,比较特殊的就是区间0/1互换这个操作,我们可以称之为异或操作
很明显我们可以知道这个性质:当一个区间被覆盖后,不管之前有没有异或标记都没有意义了
所以当一个节点得到覆盖标记时把异或标记清空
而当一个节点得到异或标记的时候,先判断覆盖标记,如果是0或1,直接改变一下覆盖标记,不然的话改变异或标记

开区间闭区间只要数字乘以2就可以处理(偶数表示端点,奇数表示两端点间的区间)
线段树功能:update:成段替换,区间异或 query:简单hash


  
  
  1. #include <cstdio>
  2. #include <cstring>
  3. #include <cctype>
  4. #include <algorithm>
  5. using namespace std;
  6. #define lson l , m , rt << 1
  7. #define rson m + 1 , r , rt << 1 | 1
  8. const int maxn = 131072;
  9. bool hash[maxn+ 1];
  10. int cover[maxn<< 2];
  11. int XOR[maxn<< 2];
  12. void FXOR(int rt) {
  13. if (cover[rt] != -1) cover[rt] ^= 1;
  14. else XOR[rt] ^= 1;
  15. }
  16. void PushDown(int rt) {
  17. if (cover[rt] != -1) {
  18. cover[rt<< 1] = cover[rt<< 1| 1] = cover[rt];
  19. XOR[rt<< 1] = XOR[rt<< 1| 1] = 0;
  20. cover[rt] = -1;
  21. }
  22. if (XOR[rt]) {
  23. FXOR(rt<< 1);
  24. FXOR(rt<< 1| 1);
  25. XOR[rt] = 0;
  26. }
  27. }
  28. void update(char op,int L,int R,int l,int r,int rt) {
  29. if (L <= l && r <= R) {
  30. if (op == 'U') {
  31. cover[rt] = 1;
  32. XOR[rt] = 0;
  33. } else if (op == 'D') {
  34. cover[rt] = 0;
  35. XOR[rt] = 0;
  36. } else if (op == 'C' || op == 'S') {
  37. FXOR(rt);
  38. }
  39. return ;
  40. }
  41. PushDown(rt);
  42. int m = (l + r) >> 1;
  43. if (L <= m) update(op , L , R , lson);
  44. else if (op == 'I' || op == 'C') {
  45. XOR[rt<< 1] = cover[rt<< 1] = 0;
  46. }
  47. if (m < R) update(op , L , R , rson);
  48. else if (op == 'I' || op == 'C') {
  49. XOR[rt<< 1| 1] = cover[rt<< 1| 1] = 0;
  50. }
  51. }
  52. void query(int l,int r,int rt) {
  53. if (cover[rt] == 1) {
  54. for ( int it = l ; it <= r ; it ++) {
  55. hash[it] = true;
  56. }
  57. return ;
  58. } else if (cover[rt] == 0) return ;
  59. if (l == r) return ;
  60. PushDown(rt);
  61. int m = (l + r) >> 1;
  62. query(lson);
  63. query(rson);
  64. }
  65. int main() {
  66. cover[ 1] = XOR[ 1] = 0;
  67. char op , l , r;
  68. int a , b;
  69. while ( ~ scanf( "%c %c%d,%d%c\n",&op , &l , &a , &b , &r) ) {
  70. a <<= 1 , b <<= 1;
  71. if (l == '(') a ++;
  72. if (r == ')') b --;
  73. if (a > b) {
  74. if (op == 'C' || op == 'I') {
  75. cover[ 1] = XOR[ 1] = 0;
  76. }
  77. } else update(op , a , b , 0 , maxn , 1);
  78. }
  79. query( 0 , maxn , 1);
  80. bool flag = false;
  81. int s = -1 , e;
  82. for ( int i = 0 ; i <= maxn ; i ++) {
  83. if (hash[i]) {
  84. if (s == -1) s = i;
  85. e = i;
  86. } else {
  87. if (s != -1) {
  88. if (flag) printf( " ");
  89. flag = true;
  90. printf( "%c%d,%d%c",s& 1? '(': '[' , s>> 1 , (e+ 1)>> 1 , e& 1? ')': ']');
  91. s = -1;
  92. }
  93. }
  94. }
  95. if (!flag) printf( "empty set");
  96. puts( "");
  97. return 0;
  98. }
练习
poj1436 Horizontally Visible Segments
poj2991 Crane
Another LCIS
Bracket Sequence

区间合并

这类题目会询问区间中满足条件的连续最长区间,所以PushUp的时候需要对左右儿子的区间进行合并
poj3667 Hotel
题意:1 a:询问是不是有连续长度为a的空房间,有的话住进最左边
2 a b:将[a,a+b-1]的房间清空
思路:记录区间中最长的空房间
线段树操作:update:区间替换 query:询问满足条件的最左断点

  
  
  1. #include <cstdio>
  2. #include <cstring>
  3. #include <cctype>
  4. #include <algorithm>
  5. using namespace std;
  6. #define lson l , m , rt << 1
  7. #define rson m + 1 , r , rt << 1 | 1
  8. const int maxn = 55555;
  9. int lsum[maxn<< 2] , rsum[maxn<< 2] , msum[maxn<< 2];
  10. int cover[maxn<< 2];
  11. void PushDown(int rt,int m) {
  12. if (cover[rt] != -1) {
  13. cover[rt<< 1] = cover[rt<< 1| 1] = cover[rt];
  14. msum[rt<< 1] = lsum[rt<< 1] = rsum[rt<< 1] = cover[rt] ? 0 : m - (m >> 1);
  15. msum[rt<< 1| 1] = lsum[rt<< 1| 1] = rsum[rt<< 1| 1] = cover[rt] ? 0 : (m >> 1);
  16. cover[rt] = -1;
  17. }
  18. }
  19. void PushUp(int rt,int m) {
  20. lsum[rt] = lsum[rt<< 1];
  21. rsum[rt] = rsum[rt<< 1| 1];
  22. if (lsum[rt] == m - (m >> 1)) lsum[rt] += lsum[rt<< 1| 1];
  23. if (rsum[rt] == (m >> 1)) rsum[rt] += rsum[rt<< 1];
  24. msum[rt] = max(lsum[rt<< 1| 1] + rsum[rt<< 1] , max(msum[rt<< 1] , msum[rt<< 1| 1]));
  25. }
  26. void build(int l,int r,int rt) {
  27. msum[rt] = lsum[rt] = rsum[rt] = r - l + 1;
  28. cover[rt] = -1;
  29. if (l == r) return ;
  30. int m = (l + r) >> 1;
  31. build(lson);
  32. build(rson);
  33. }
  34. void update(int L,int R,int c,int l,int r,int rt) {
  35. if (L <= l && r <= R) {
  36. msum[rt] = lsum[rt] = rsum[rt] = c ? 0 : r - l + 1;
  37. cover[rt] = c;
  38. return ;
  39. }
  40. PushDown(rt , r - l + 1);
  41. int m = (l + r) >> 1;
  42. if (L <= m) update(L , R , c , lson);
  43. if (m < R) update(L , R , c , rson);
  44. PushUp(rt , r - l + 1);
  45. }
  46. int query(int w,int l,int r,int rt) {
  47. if (l == r) return l;
  48. PushDown(rt , r - l + 1);
  49. int m = (l + r) >> 1;
  50. if (msum[rt<< 1] >= w) return query(w , lson);
  51. else if (rsum[rt<< 1] + lsum[rt<< 1| 1] >= w) return m - rsum[rt<< 1] + 1;
  52. return query(w , rson);
  53. }
  54. int main() {
  55. int n , m;
  56. scanf( "%d%d",&n,&m);
  57. build( 1 , n , 1);
  58. while (m --) {
  59. int op , a , b;
  60. scanf( "%d",&op);
  61. if (op == 1) {
  62. scanf( "%d",&a);
  63. if (msum[ 1] < a) puts( "0");
  64. else {
  65. int p = query(a , 1 , n , 1);
  66. printf( "%d\n",p);
  67. update(p , p + a - 1 , 1 , 1 , n , 1);
  68. }
  69. } else {
  70. scanf( "%d%d",&a,&b);
  71. update(a , a + b - 1 , 0 , 1 , n , 1);
  72. }
  73. }
  74. return 0;
  75. }

练习
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值