[线段树系列] 普通线段树

线段树是一种强大的数据结构，用于维护区间、图、树等各类数据。

线段树的“强大”体如今它面对各类类型的数据都有应付的方式，并且不断有“改进”版线段树的产生。

线段树是基于递归和分治思想的数据结构，通常用于维护具备“区间可加性”的数据。

什么是”区间可加性“呢，举几个例子：

区间和，区间最大最小值，区间LCA，区间质数个数

这些东西都有共同的特性：f(x,y)=f(f(x,z),f(z,y))，z∈[x,y]

因而咱们就能够用线段树来维护。

随手画了张线段树的图，它大概长这样：

是否是很神奇?

它是怎么维护数据的呢？( 以维护区间数据为例 )

让咱们用区间最大值为例：

假设原数组a是{1,2,3,4,5,6} (我习惯下标从1开始)

咱们把原数组插入线段树，看看它各个节点的值：

假设咱们要查询区间 [2,4] 的值：

咱们发现并无 [2,4] 这个节点，那怎么查询呢？

咱们计算出 [2,4] 的mid值，mid=(l+r)/2=3。

而后咱们查询区间 [2,3] 和区间 [4,4] ( 即区间[l,mid]和区间[mid+1,r] )。

区间 [4,4] 是叶子节点，返回它的值4。

回到 [2,3] 咱们继续递归计算max( [2,2],[3,3] )返回3

获得最后的最大值4

这一段的代码：

int query(int p,int l,int r){ if(l<=l(p) && r>=r(p))return val(p); int mid=(l(p)+r(p))>>1; int val=-INF; if(l<=mid)val=max(val,query(p<<1,l,r)); if(r>mid)val=max(val,query(p<<1|1,l,r)); return val; }

等等，既然要维护数据，怎么能不支持修改呢？

线段树支持单点修改和区间修改。

单点修改很是简单，我这里就不介绍了。

直接来区间修改。

咱们每次更新一个值都要把全部包含它的节点更新一次。

可是若是更新一整个区间，这个更新量是很是大的，做为一种高效数据结构是不会容许这么慢的修改的。

因而咱们引进一个”打标记“的思想。

也就是给咱们要更新的值先打上”你已经被修改了“的标记，等咱们要用它的时候在改它。

在线段树中被称为”下发懒标记“——pushdown lazytag

下传标记的代码( 以维护区间和为例 )：

void pushdown(int p){ if(add(p)){ val(p<<1)+=add(p)*(r(p<<1)-l(p<<1)+1); val(p<<1|1)+=add(p)*(r(p<<1|1)-l(p<<1|1)+1); add(p<<1)+=add(p); add(p<<1|1)+=add(p); add(p)=0; } }

这里咱们的懒标记是加(add)，懒标记还能够是乘(mul)，或者加乘混合(须要符合运算法则)

接下来放建树的代码：

void pushup(int p){ val(p)=val(p<<1)+val(p<<1|1); } void build(int p,int l,int r){ l(p)=l,r(p)=r;//保存每一个节点的左右儿子的编号
    if(l==r){ val(p)=a[l]; return; }pushdown(p); int mid=(l+r)>>1; build(p<<1,l,mid);build(p<<1|1,mid+1,r);//递归建树 
    pushup(p);//更新值
}

上一份维护区间和的完整代码吧：

 

#include<bits/stdc++.h>
using namespace std; inline int read(){ int data=0,w=1;char ch=0; while(ch!='-' && (ch<'0'||ch>'9'))ch=getchar(); if(ch=='-')w=-1,ch=getchar(); while(ch>='0' && ch<='9')data=data*10+ch-'0',ch=getchar(); return data*w; }//快读
const int maxn=100010; int n,m,a[maxn]; struct SegmentTree{ int l,r;//左右节点编号
    int val,add;//节点值，标记
    #define l(x) tree[x].l
    #define r(x) tree[x].r
    #define val(x) tree[x].val
    #define add(x) tree[x].add }tree[maxn<<2]; void pushup(int p){ val(p)=val(p<<1)+val(p<<1|1);//更新，一个节点用它的左右两个节点更新
} void build(int p,int l,int r){//建树代码
    l(p)=l,r(p)=r; if(l==r){ val(p)=a[l];return; } int mid=(l+r)>>1; build(p<<1,l,mid);build(p<<1|1,mid+1,r); pushup(p); } void pushdown(int p){//下传懒标记
    if(add(p)){ val(p<<1)+=add(p)*(r(p<<1)-l(p<<1)+1); val(p<<1|1)+=add(p)*(r(p<<1|1)-l(p<<1|1)+1); add(p<<1)+=add(p); add(p<<1|1)+=add(p); add(p)=0; } } int query(int p,int l,int r){//查询
    if(l<=l(p) && r>=r(p))return val(p); pushdown(p); int mid=(l(p)+r(p))>>1; int ret=0; if(l<=mid)ret+=query(p<<1,l,r); if(r>mid)ret+=query(p<<1|1,l,r); return ret; } void update(int p,int l,int r,int d){//修改
    if(l<=l(p) && r>=r(p)){ val(p)+=d*(r(p)-l(p)+1);add(p)+=d; return; } pushdown(p); int mid=(l(p)+r(p))>>1; if(l<=mid)update(p<<1,l,r,d); if(r>mid)update(p<<1|1,l,r,d); pushup(p); } int main(){ n=read();m=read();//读入数据个数，操做个数
    for(int i=1;i<=n;i++)a[i]=read();//读入数据
    build(1,1,n);//建树
    int opt,l,r,d; while(m--){ opt=read();l=read();r=read(); if(opt==1)//查询
            printf("%d\n",query(1,l,r)); else{ d=read();//区间加上这个值
            update(1,l,r,d);//修改
 } } return 0; }

 

撰文不易，但愿能帮到各位，下一篇讲动态开点线段树，本系列持续更新