[笔记]浅谈线段树

（被某D姓大佬嘲讽了。。于是乎安利一发大佬的树状数组：https://www.luogu.org/blog/ddy/qian-tan-shu-zhuang-shuo-zu-post）

 

---

 

 线段树是个好东西

这篇文章主要是讲一讲

基础，自己的心得；

好，那就开始！

一.定义

线段树（Segment Tree）是一棵

二叉搜索树
树上每个节点代表一个区间[ l , r ]
同一层的节点所代表的区间，相互不会重叠

同一层节点所代表的区间，加起来是个连续的区间

对于每一个非叶结点所表示的结点 [ l , r ]，

其左儿子表示的区间为 [ l , ( l + r ) / 2 ] （[ l , ( l + r ) >>1 ]）

右儿子表示的区间为 [ ( l + r ) / 2 + 1, r ] (除法去尾取整) ( [ ( l + r ) >> 1 + 1 , r ] )

叶子节点表示的区间长度为1

 

ps: a * 2 = a << 1 ;

a * 2 + 1 = a << 1 | 1 ;
(下文 全部用位运算替代)

#define ll long long
struct node{
ll l,r,sum,add,mul;
}t[M<<2];

M为数列元素数 （M<<2=M*4），l是左端点，r是右端点，sum为数列区间[ l , r ]中所有元素总和，add为加法标记，mul为乘法标记（与lazy是一个意思，不懂的后面会讲）

开四倍空间证明

证明：
设共有N个节点，则有

ceil（log2（n））层，

所以共需要2^(ceil(log2(n))+1)-1节点

 

 个人理解：其实就是将一个数列不断二分，直到区间只有一个叶子节点。而每个节点的父节点具有在一定情况下能代表其所有子节点的性质（比如求和时，t[a].sum=t[a<<1].sum+t[a<<1|1].sum），从而通过访问父节点，得到性质，节省访问子节点的时间，降低时间复杂度

如图

二.操作

　　1.建树

　　　　void build(根节点（一般是1），左端点，右端点)

　　　　先二分区间（层层递归），直至区间左端点等于右端点，这意味着到达叶子节点，所以读取数据；

　　　　　最后递归回去，求得区间的sum值。

void build(ll tr,ll l,ll r)
{
    t[tr].l=l,t[tr].r=r,t[tr].mul=1;
    if(l==r)
    {
        t[tr].sum=g();  //inline ll g() 快读
        return ;
    }
    R ll mid=(l+r)>>1;
    build(tr<<1,l,mid);
    build(tr<<1|1,mid+1,r);
    t[tr].sum=t[tr<<1].sum+t[tr<<1|1].sum;
}

 

 

 

　　2.改变元素的值

　　　　（1）加法

　　　　　　void add（根节点，左端点，右端点，增量）

　　　　　　在现有区间（这是由根节点决定的->相当于通过根确定子树->从而确定区间），二分区间位置　　

　　　　　　如果递归时，现有区间就是所要更改的区间，那么直接更改整个区间的sum，使sum+=inc*（r-l+1）（就是增量*区间长度），同时add+=inc（不能是add=inc，因为add可能不为0），add表示的是[l,r]区间中所有的叶子节点都少加了add，而之所以要有这个add是为了减少不必要的操作（此时没必要把每个子节点的值都修改）（其实原因就是上文：每个节点的父节点具有在一定情况下能代表其所有子节点的性质，从而通过访问父节点，得到性质，节省访问子节点的时间，降低时间复杂度）

　　　　　　　　　　　　而所要更改的区间的左端点如果大于现有区间的mid，则递归右半区间；

　　　　　　　　　　　　而所要更改的区间的右端点如果小于现有区间的mid，则递归左半区间；

　　　　　　　　　　　　而所要更改的区间就在现有区间内的的话，分别递归区间[l,mid]和[mid+1,r];

　　　　　　　　　　　　细心的你（？）可能发现，为什么在后三种情况递归前，会有一些add(),和mul()运算？

　　　　　　　　　　　　因为我们记得 “每个节点的父节点具有在一定情况下能代表其所有子节点的性质” 对吧？ 我们也知道，父节点的sum是由子节点递归而来的吧？ 所以如果不下放add（此时这个操作是必要的），那么父节点的sum不能被更新，于是就不能代表子节点的性质了

如图：

所以实际上是为了保持父节点的代表性。。

 

void add(ll tr,ll l,ll r,ll inc)
{
    if(t[tr].l==l&&t[tr].r==r)
    {
        t[tr].sum+=inc*(r-l+1);
        t[tr].add+=inc;
        return ;
    }
    R ll mid=(t[tr].l+t[tr].r)>>1;
    if(t[tr].mul!=1) 
        mul(tr<<1,t[tr].l,mid,t[tr].mul),mul(tr<<1|1,mid+1,t[tr].r,t[tr].mul),t[tr].mul=1;
    if(t[tr].add!=0)
        add(tr<<1,t[tr].l,mid,t[tr].add),add(tr<<1|1,mid+1,t[tr].r,t[tr].add),t[tr].add=0;
    if(l>mid) add(tr<<1|1,l,r,inc);
    else if(r1) add(tr<<1,l,r,inc);
    else add(tr<<1,l,mid,inc),add(tr<<1|1,mid+1,r,inc);
    t[tr].sum=t[tr<<1].sum+t[tr<<1|1].sum;
}

 

 

　　　　（2）乘法

　　　　　　void mul（根节点，左端点，右端点，增量）

　　　　　　你会发现，递归的过程基本相同；

　　　　　　　　那我需要解释一下，为什么先乘后加

　　　　　　　　若对[l,r]进行+1，*2，+1的操作

　　　　　　　　则实际上是（a+1）*2 + 1=2*a+2 + 1；

　　　　　　　　所以mul要sum*=inc，mul*=inc，add*=inc

　　　　　　　　所以先前的add要乘mul，再进行add

　　　　　　　　

void mul(ll tr,ll l,ll r,ll inc)
{
    if(t[tr].l==l&&t[tr].r==r)
    {
        t[tr].sum*=inc;
        t[tr].mul*=inc;
        t[tr].add*=inc;
        return ;
    }
    R ll mid=(t[tr].l+t[tr].r)>>1;
    if(t[tr].mul!=1) 
        mul(tr<<1,t[tr].l,mid,t[tr].mul),mul(tr<<1|1,mid+1,t[tr].r,t[tr].mul),t[tr].mul=1;
    if(t[tr].add!=0)
        add(tr<<1,t[tr].l,mid,t[tr].add),add(tr<<1|1,mid+1,t[tr].r,t[tr].add),t[tr].add=0;
    if(l>mid) mul(tr<<1|1,l,r,inc);
    else if(r1) mul(tr<<1,l,r,inc);
    else mul(tr<<1,l,mid,inc),mul(tr<<1|1,mid+1,r,inc);
    t[tr].sum=t[tr<<1].sum+t[tr<<1|1].sum;
}

 

　　　

 

　　　3.求和

 

　　　　　ll query(根节点，左端点，右端点)

　　　　你又会发现，递归的过程同add

　　　　　所以理解后就很好懂啦。

ll query(ll tr,ll l,ll r)
{
    if(t[tr].l==l&&t[tr].r==r) return t[tr].sum;
    R ll mid=(t[tr].l+t[tr].r)>>1;
    if(t[tr].mul!=1) 
        mul(tr<<1,t[tr].l,mid,t[tr].mul),mul(tr<<1|1,mid+1,t[tr].r,t[tr].mul),t[tr].mul=1;
    if(t[tr].add!=0)
        add(tr<<1,t[tr].l,mid,t[tr].add),add(tr<<1|1,mid+1,t[tr].r,t[tr].add),t[tr].add=0;
    if(l>mid) return query(tr<<1|1,l,r);
    else if(r1) return query(tr<<1,l,r);
    else return query(tr<<1,l,mid)+query(tr<<1|1,mid+1,r);
}

 

 

 

 三.例题

　　如洛谷P3372 P3373（说实话，我的线段树不是最快的，但还可以）（我太菜了，只做了这些）

　

其实我的思路就是

维护线段树的性质，该更新时则更新，通过访问父节点，得到性质，节省访问子节点的时间，降低时间复杂度。

 

如有错误，恳请您指正（我太菜了）；如有不理解，可留言，我会尽量回复。。。（高中生吐槽一波。。）

by Jackpei 2019.2.5

转载于:https://www.cnblogs.com/Jackpei/p/10352817.html