数据结构之线段树(Segment Tree)

线段树的概念

线段树(Segment Tree)也是一棵树，只不过元素的值代表一个区间。
常用区间的 统计 操作，比如一个区间的最大值(max)，最小值(min)，和(sum)等等。

线段树是一种平衡二叉搜索树(完全二叉树)，它将一个线段区间划分成一些单元区间。对于线段树中的每一个非叶子节点[a,b]，它的左儿子表示的区间为[a,(a+b)/2]，右儿子表示的区间为[(a+b)/2+1,b]，最后的叶子节点数目为N，与数组下标对应。线段树 的一般包括建立、查询、插入、更新等操作，建立规模为N的时间复杂度是 O(NlogN)，其他操作时间复杂度为O(logN)。

 

分段树是完全二叉树，可以用一维数组存储。

如果我们用数组来存储线段树的话，我们大致需要开辟多大的数组空间呢？

根据而前面我们对满二叉树的分析

h层的满二叉树总共有 2^h-1 个节点，第h-1层有2^(h-1)个节点，它们大概是两倍的关系。

也就是说对于满二叉树 最后一层的节点数乘以2 大致就是整棵树的节点数。

但是线段树并不一定是满二叉树，但是一定是平衡二叉树，所以需要多冗余一层。也就是 乘以4 就足以盛放所有的节点数，但是会浪费一定的内存空间。

 

线段树的基本操作

构建线段树

根据上面我们对线段树的描述，构建一个线段树就比较简单了，根节点就是整个区间，根节点的左右子树平分根节点的区间，直至区间内只剩下一个元素不能平分为止。

修改线段树

针对上面的数组，把索引为 2 的值改成 10 如下图所示

线段树的查询

对于线段树的查询，主要有以下几种情况：

• 要查询的区间在刚好就是当前节点的区间
• 要查找的区间在当前节点的左子树区间
• 要查找的区间在当前节点的右子树区间
• 要查找的区间一部分在当前节点的左子树区间，一部分在右子树区间

 

实现一个线段树

下面实现的线段树，有三个功能：

• 把数组构建成一颗线段树
• 线段树的修改
• 线段树的查询

/**
 * @author yangjunpeng
 * @version $Id: ArraySegmentTree.java, v 0.1 2020年04月08日 下午10:38 yangjunpeng Exp