《林林数据结构笔记》线段树求数组区间和，单点更新，区间更新+lazy思想

昨天刷力扣认识的这个数据结构，原题在这里。记录一下力扣https://leetcode.cn/problems/my-calendar-iii/

例题

给定数组[1, 2, 3, 4, 5, ....]，编写函数来返回数组任意区间元素的和

朴素的想法

直接把每个元素相加，时间复杂度为O(n)，n是区间长度

def getSum(arr, start, end):
    val = 0
    for i in range(start, end + 1):
        val += arr[i]
    return val

前缀和算法

前缀和是一种常用的、较为高效的预处理方式。能够有效降低查询的时间复杂度。前缀和可以理解为数组前项的和。查询的时间复杂度为O(1)

class PartialSumArr:
    def __init__(self, arr):
        val = 0
        self.partialSumArr = []
        for i in arr:
            val += i
            self.partialSumArr.append(val)

    def getSum(self, start, end):
        if start == 0:
            return self.partialSumArr[end]
        return self.partialSumArr[end] - self.partialSumArr[start-1]


partialSumArr = PartialSumArr([1, 2, 3, 4, 5, 6, 7])
print(partialSumArr.partialSumArr)
print(partialSumArr.getSum(0, 4))
print(partialSumArr.getSum(3, 6))


>>>[1, 3, 6, 10, 15, 21, 28]
15
22

新的问题

如果要修改数组中某一项，即更新数组。

对于方法一：更新的时间复杂度为O(1)；

对于方法二：更新的时间复杂度为O(n)，因为需要把前缀和数组中这个元素之后每个元素都更新一遍

如果要实现区间更新呢？

对于方法一：更新的时间复杂度为O(n)；n为区间大小

对于方法二：更新的时间复杂度为O(m*n),m为区间最右下标位置距离数组最右距离，n为区间长度

线段树

线段树是一种数据结构，可以有效地对数组进行范围内的查询，同时可以足够灵活地修改数组。

给定一个数组(arr)，编写函数来求其任意区间元素和

建树

这是根节点

 通过求两个下标的平均值来确定其子节点

同理

 

 

 

至此我们的树就建立好了，下面创建一个数组(nodes)来储存树的节点。当父节点为node[index]时，我们把它的子节点放在node[index * 2]和node[index * 2 + 1]位置。我们通常把根节点放在下标1的位置。如下图，数组里0, 4表示下标0～4的元素和，这么写方便和上面树图对比

 

替换成对应值

对应代码:

class SegmentTreeSum:
    def __init__(self, arr):
        self.arr = arr
        self.nodes = [0] * len(arr) * 2
        self.construct(1, 0, len(arr) - 1)

    de