1.定义
线段树是一种基于分治的数据结构,用来进行快速的区间查询、单点修改操作(O(logn));
线段树中每个结点都用于存储一段区间相关的信息(如区间和、区间最值等);
2.性质
(1)线段树是BST、AVL树,不是完全二叉树
(2)线段树可以使用一维数组存储,此时结点所对应的区间、结点之间的关系均由数组下标隐含表示:
设从下标1开始按层次依次存储线段树的结点,
若左孩子存在,则:lchild_index=p_index*2
若右孩子存在,则:rchild_index=p_index*2+1(3)线段树中不存在度为1的结点
证明:很明显除叶结点之外,其它结点对应的区间均可以分成两半
(4)线段树的每层中,所有结点对应的区间互不相交
证明:
I. 对于兄弟结点,它们所对应的区间一定互不相交;
II.对于堂兄弟结点,由“I.”得:它们的共同祖先对应的左右孩子所对应的区间一定不相交,故它们自身对应的区间也一定互不相交;
(3)线段树的所有叶结点对应的区间长度为1,它们从左往右组成整个区间
3.实现(用线段树维护区间和)
(1)数据结构
#include<iostream>
#include<vector>
class SegTree{
private:
std::vector<int> ST;
public:
SegTree(const std::vector<int> &arr);
void build(int index, int l, int r, const std::vector<int> &arr);
void find(int index, int l, int r, int start, int end);
void singleChange(int index, int l, int r);
};
(2)建树(分治+DFS 后序)
SegTree::SegTree(const std::vector<int> &arr){
ST=vector<int>(arr.size()*4+1);
build(1, 0, arr.size()-1, arr);
}
// 初始时,index=1、[l,r]=[0, arr.size()-1]
void SegTree::build(int index, int l, int r, const std::vector<int> &arr){
if(l==r){
ST[index]=arr[l];
return;
}
int mid = (l+r)/2;
build(index*2, l, mid, arr);
build(index*2+1, mid+1, r, arr);
ST[index] = ST[index*2] + ST[index*2+1];
}
(3)区间查询
区间查询有三种情况需要考虑(设待查区间为[start, end],当前区间为[l,r]):
当前区间的中点为 mid = (l+r)/2;
I. [start, end]位于当前区间的左半部,即 end <= mid,转左子树继续查询;
II. [start, end]位于当前区间的右半部,即 start > mid,转右子树继续查询;
III. [start, end]包含当前区间的中点,即 start <= mid && end > mid,将[start, end]分为[start, mid]、[mid+1, end],并分别转左、右子树继续查询,然后将查询结果合并即可(这里的线段树维护的是区间和,故合并操作就是将两个区间查询结果加起来);
// 初始时,index=1、[l,r]=[0,arr.size()-1]
int SegTree::find(int index, int l, int r, int start, int end){
if(start==l && end==r){
return ST[index];
}
int mid = (l+r)/2;
if(end <= mid){
return find(index*2, l, mid, start, end);
}
if(start>mid){
return find(index*2+1, mid+1, r, start, end);
}
return find(index*2, l, mid, start, mid) + find(index*2+1, mid+1, r, mid+1, end);
}
(4)单点修改
I. 要使arr[target]=value,首先需要将线段树中对应叶结点所维护的区间和修改为value;
II. 因性质(4)“线段树的每层中,所有结点所对应的区间互不相交”,故对某个叶结点修改,只会影响到其祖先结点
III.所以完整的单点修改操作,除了修改叶结点,还需要从叶结点开始,自底向上修改其所有祖先结点,可以用后序遍历实现
// 初始时,index=1、[l,r]=[0,arr.size()-1]
void SegTree::singleChange(int index, int l, int r, int target, int value){
if(l == r){
ST[index] = value;
return;
}
int mid = (l+r)/2;
if(target <= mid){
singleChange(index*2, l, mid, target, value);
}else{
singleChange(index*2+1, mid+1, r, target, value);
}
ST[index] = ST[index*2] + ST[index*2+1];
}