线段树【C++实现】

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];
}