数据结构--线段树

什么是线段树？

线段树是一种二叉树，广义上也被归类为二叉搜索树。与区间树类似，将一个区间划分成一些单元区间，每个单元区间(区间长度为1)对应线段树中的一个叶结点。

对于区间的修改、维护和查询时间复杂度为O(logN)级别，实际应用时一般还要开4N的数组以免越界。

线段树的局限性:必须符合区间加法。区间加法是指对于[l,r]的区间，它的答案可以由[l,m]和[m+1,r]的答案合并求出。

符合区间加法的例子：

区间之和——总区间之和 = 左区间之和 + 右区间之和

最大公因数(GCD)——总GCD = gcd( 左区间GCD , 右区间GCD );

最大值——总最大值=max(左区间最大值，右区间最大值)

不符合区间加法的例子：

众数——只知道左右区间的众数，没法求总区间的众数

01序列的最长连续零——只知道左右区间的最长连续零，没法知道总的最长连续零

1、建树

以堆的方式存储数据，用数组来模拟树形结构，对于对于每一个节点R ,左子节点为 2*R (位运算R<<1)右子节点为 2*R+1（位运算R<<1|1，以1为根节点。

#define maxn 100007  //元素总个数  
#define ls l,m,rt<<1  
#define rs m+1,r,rt<<1|1  

int Sum[maxn<<2],Add[maxn<<2];//Sum求和，Add为懒惰标记   
int A[maxn],n;//存原数组数据下标[1,n] 

//PushUp函数更新节点信息 ，这里是求和  
void PushUp(int rt) { 
    Sum[rt] =Sum[rt<<1]+Sum[rt<<1|1];
}  

//Build函数建树   
void Build(int l,int r,int rt){ //l,r表示当前节点区间，rt表示当前节点编号  
    if(l==r) {//若到达叶节点   
        Sum[rt]=A[l];//储存数组值   
        return;  
    }  
    int m=(l+r)>>1;  
    //左右递归   
    Build(l,m,rt<<1);  
    Build(m+1,r,rt<<1|1);  
    //更新信息   
    PushUp(rt);  
}  

int main()
{
    //建树   
    Build(1,n,1);  
    return 0
}

2、单点修改/区间修改  <*lazy标记>

单点修改，假设A[L]+=C:

//调用点修改 Update(L,C,1,n,1);  
void Update(int L,int C,int l,int r,int rt){//l,r表示当前节点区间，rt表示当前节点编号  
    if(l==r){//到叶节点，修改   
        Sum[rt]+=C;  
        return;  
    }  
    int m=(l+r)>>1;  
    //根据条件判断往左子树调用还是往右   
    if(L <= m) Update(L,C,l,m,rt<<1);  
    else       Update(L,C,m+1,r,rt<<1|1);  
    PushUp(rt);//子节点更新了，所以本节点也需要更新信息   
}   

区间修改,假设A[L,R]+=C：

//调用区间修改 Update(L,R,C,1,n,1);  
void Update(int L,int R,int C,int l,int r,int rt){//L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号   
    if(L <= l && r <= R){//如果本区间完全在操作区间[L,R]以内   
        Sum[rt]+=C*(r-l+1);//更新数字和，向上保持正确  
        Add[rt]+=C;//增加Add标记，表示本区间的Sum正确，子区间的Sum仍需要根据Add的值来调整  
        return ;   
    }  
    int m=(l+r)>>1;  
    PushDown(rt,m-l+1,r-m);//下推标记  
    //这里判断左右子树跟[L,R]有无交集，有交集才递归   
    if(L <= m) Update(L,R,C,l,m,rt<<1);  
    if(R >  m) Update(L,R,C,m+1,r,rt<<1|1);   
    PushUp(rt);//更新本节点信息   
}   

3、区间查询

询问A[L,R]的和,首先是下推标记的函数：

void PushDown(int rt,int ln,int rn){  
    //ln,rn为左子树，右子树的数字数量。   
    if(Add[rt]){  
        //下推标记   
        Add[rt<<1]+=Add[rt];  
        Add[rt<<1|1]+=Add[rt];  
        //修改子节点的Sum使之与对应的Add相对应   
        Sum[rt<<1]+=Add[rt]*ln;  
        Sum[rt<<1|1]+=Add[rt]*rn;  
        //清除本节点标记   
        Add[rt]=0;  
    }  
}   
//调用区间查询 int ANS=Query(L,R,1,n,1);  
int Query(int L,int R,int l,int r,int rt){//L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号  
    if(L <= l && r <= R){  
        //在区间内，直接返回   
        return Sum[rt];  
    }  
    int m=(l+r)>>1;  
    //下推标记，否则Sum可能不正确  
    PushDown(rt,m-l+1,r-m);   
      
    //累计答案  
    int ANS=0;  
    if(L <= m) ANS+=Query(L,R,l,m,rt<<1);  
    if(R >  m) ANS+=Query(L,R,m+1,r,rt<<1|1);  
    return ANS;  

参考:线段树详解 （原理，实现与应用）

视频讲解 线段树