线段树入门总结

来源:互联网 发布:sql 求平均值 编辑:程序博客网 时间:2024/05/01 06:56

树结构的基本思想是分割,普通二叉搜索树是按照数据来划分(想了解二叉搜索树的请移步:Here),线段树处理的对象是线段(区间也可以看成线段,L==R时为一个点),它把线段组织成有利于检索和统计的形式,它的本质是线段的二叉搜索树。但是它的线段可以分解和合并,线段树又有一些一般二叉检索树没有的特殊操作。另外线段树操作的是整个区间,它的时间复杂度不依赖于数据对象。它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。
线段树的定义:
线段树是一棵二叉检索树,最终将一个区间 [1,n] 划分为一些 [ i,i+1 ]的但愿区间,每个单元区间对应一个叶节点。叶节点区间只有一个数(L == R)。对于每个线段树的非叶子节点 [ a, b ],其左儿子 为 [a,(a+b)/2],右儿子为 [ (a+b)/2 + 1,b]。所以线段树是一颗平衡二叉树。最后节点数目为 N 即整段区间的长度,其节点数为 n + n/2 + n/4 + … = 2n。所以其空间复杂度为 O(2n)。
时间复杂度为 log n(插入或者读取均为log n)。其结构如图
线段树的处理对象:
 线段树的处理对象是一段区间,区间上的格点对应有限个区域的变量,处理问题的时候,抽出区间上的格点,也是明确每个格点对应变量的含义。
线段树的基本结构:
一棵线段树,应该具有 插入 , 修改 , 查询 三个基本功能。
【代码实现】:
我们以区间最值问题(RMQ)为例:
我们先声明线段树所需的结构体
[cpp] view plain copy

#include <stdio.h>  #define MAXN 1<<19  typedef struct  {      int value;      //区间最值      int left,right; //区间范围  }Tree;  Tree node[2*MAXN];  int father[MAXN];   //记录叶子对应结构体的 下标</span>  

如图所示:
如果 二叉树 的 父节点 为 k,那么其 左儿子为 2k,右儿子 为 2k+1。
现在我们可以建立线段树里。
[cpp] view plain copy

//线段树的建立  void build(int i, int left, int right){    //i为结构体数组的下标      node[i].left = left;          //为节点成员初始化      node[i].right = right;      node[i].value = 0;      if(left == right){   //当线段树的节点为叶子时,结束递归          father[left] = i;//将叶子在结构体数组的下标记录,以便更新是可以自下而上          return ;      }      //现在分别建立该节点的左右孩子      build(i<<1,left,(left+right)/2);      build( (i<<1)+1,1+(left+right)/2,right);      return ;  }  

因为我们现在是以最值得问题作为样例,所以我们更新数据的话就是单点操作,这里最值我们以最大值为例
[cpp] view plain copy

//自上往下的更新,n_i 如上图所意  void Updata(int n_i){      if(n_i == 1) return ;   //找到了根节点,结束递归      int fa = n_i/2;     //找到了父节点      int a = node[2*fa].value;  //该父节点的左儿子的值      int b = node[2*fa + 1].value;//该父节点的右儿子的值      node[fa].value = a>b?a:b;       //更新节点数据      Updata(fa);             //递归更新      return ;  }  

现在我们剩下的就差查询操作了
[cpp] view plain copy

int Max = -MAXN;  //k为结构体下标,通常我都从根节点开始查询,所以,通常我们初始化时为1  //查询区间为 [ left, right ]  void Query(int k,int left,int right){      //当查询区间完全重合时      if(node[k].left == left && node[k].right == right){          Max = Max > node[k].value ? Max : node[k].value;          return ;      }      //对左子树进行操作      if(left <= node[2*k].right){  //如果与左区间有交集          if(right <= node[2*k].right)  //如果完全包含于左区间,则查询范围不变              Query(2*k,left,right);          else//否则这将区间查分开,先查询左边的              Query(2*k,left,node[2*k].right);      }      //对右子树进行操作      if(right >= node[2*k+1].left){  //如果与右区间有交集          if(left >= node[2*k+1].left)  //如果完全包含于右区间,则查询范围不变              Query(2*k+1,left,right);          else//否则这将区间查分开,先查询右边的              Query(2*k+1,node[2*k+1].left,right);      }      return ;  }  
0 0
原创粉丝点击