线段树的懒标记与应用

目录

一、前言

二、Lazy-tag技术

1、update() 中的lazy-tag

三、例题

1、区间修改、区间查询（lanqiaoOJ 1133）

---

一、前言

本文主要讲了线段树的Lazy-tag技术和一道例题，建议自己要多练习线段树的题目。 

二、Lazy-tag技术

• 背景：区间修改
• 简单区间修改，例如对一个数列的 [L, R] 区间内每个元素统一加上 d。
• 如果在线段树上用 “单点修改” 一个个修改这些元素，一次区间修改O(nlogn)，还不如直接暴力修改的 O(n)。
• 解决办法，利用线段树的特征：线段树的结点 tree[i]，记录了 i 这个区间的值。
• 再定义一个 tag[i]，用它统一记录 i 这个区间的修改，而不是一个个修改区间内的每个元素。
• 这个办法被称为 “lazy-tag”。一次区间修改 O(logn)。

• lazy-tag (懒惰标记，或者延迟标记)。
• 修改一个线段区间时，只对这个线段区间进行整体修改 (把修改记录在子树的根结点上)，其内部每个元素的内容先不做修改。
• 当这个线段区间的一致性被破坏时，才把变化值传递给下一层。
• 复杂度：每次区间修改的复杂度是 O(logn) 的。

1、update() 中的lazy-tag

例：把 [4, 9] 区间内的每个元素加 3：

1）左子树递归到结点 5，即区间 [4, 5]，完全包含在 [4, 9] 内，打标记 tag[5] = 3，更新 tree[5] 为 20，不再继续深入；

2）左子树递归返回，更新 tree[2] 为 30；

3）右子树递归到结点 6，即区间 [6, 8]，完全包含在 [4, 9] 内，打标记 tag[6]=3，更新 tree[6] 为 23。

4）右子树递归到结点 14，即区间 [9, 9]，打标记 tag[14]=3，更新 tree[14]=13；

5）右子树递归返回，更新 tree[7]=20；继续返回，更新 tree[3]=43；

6）返回到根结点，更新 tree[1]=73。

【多次修改】

• 如果发生多次修改同一个结点上的 tag 会发生多次修改，导致冲突。
• 例如做 2 次区间修改，一次是 [4, 9]，一次是 [5, 8]，它们都会影响 5:[4, 5] 这个结点。

第一次修改 [4, 9] 覆盖了结点 5，用 tag[5] 做了记录；

第二次修改 [5, 8] 不能覆盖结点 5，需要再向下搜到结点 11:[5, 5]，从而破坏了 tag[5]，此时原 tag[5] 记录的区间统一修改就不得不往它的子结点传递和执行了，传递后 tag[5] 失去了用途，需要清空。

【push_down】

lazy-tag 的主要操作是解决多次区间修改的冲突，用 push_down() 函数完成。它首先检查结点 p 的 tag[p]，如果有值，说明前面做区间修改时给 p 打了 tag 标记，接下来就把 tag[p] 传给左右子树，然后把 tag[p] 清零。push_down() 函数不仅在 “区间修改” 中用到，在 “区间查询” 中同样用到。

三、例题

1、区间修改、区间查询（lanqiaoOJ 1133）

注：该题用树状数组讲过

【题目描述】

给定一个长度为 N 的数组 a，初值为 a1, a2, ..., aN

有 Q 个操作，操作有两种：

1 L R d：将区间 [L,R] 内每个数加上 d。

2 L R：输出区间 [L, R] 内每个数的和。

【输入描述】

第 1 行是整数 N、M，表示数组 a 的长度和操作个数。1<=n, m<=10^5

第 2 行包含 N 个非负整数 a1, a2, ..., aN，表示数组 a 的初值

第 3~M-2 行每行表示一个操作

【输出描述】

输出每行一个整数，表示查询的答案

【输入样例】

5 5

1 2 3 4 5

2 1 2

1 2 3 1

2 1 3

1 1 5 1

2 1 5

【输出样例】

3

8

22

【完整代码】

def bulid(p,pl,pr):     #建树
    if pl==pr:
        tree[p]=a[pl]
        return
    mid=(pl+pr)>>1
    bulid(p<<1,pl,mid)
    bulid(p<<1|1,mid+1,pr)
    tree[p]=tree[p<<1]+tree[p<<1|1]     #push_up(p)

def addtag(p,pl,pr,d):       #给结点p打tag标记,并更新tree
    tag[p]+=d               #打上tag标记
    tree[p]+=d*(pr-pl+1)    #计算新的tree

def push_down(p,pl,pr):
    if tag[p]>0:            #有tag标记，这是以前做区间修改时留下的
        mid=(pl+pr)>>1
        addtag(p<<1,pl,mid,tag[p])      #把tag标记传给左子树
        addtag(p<<1|1,mid+1,pr,tag[p])  #将tag标记传给右子树
        tag[p]=0                        #p自己的tag被传走了

def update(L,R,p,pl,pr,d):
    if L<=pl and pr<=R:     #在这个区间里面
        addtag(p,pl,pr,d)
        return
    push_down(p,pl,pr)  #将懒惰标记传递给孩子.tree[p]不能完全被包含在需要修改的区间[L,R]中，需要解决多次修改的冲突问题
    mid=(pl+pr)>>1
    if L<=mid:
        update(L,R,p<<1,pl,mid,d)
    if R>=mid+1:
        update(L,R,p<<1|1,mid+1,pr,d)
    tree[p]=tree[p<<1]+tree[p<<1|1]     #push_up(p)

def query(L,R,p,pl,pr):
    if L<=pl and pr<=R:
        return tree[p]
    push_down(p,pl,pr)
    res=0
    mid=(pl+pr)>>1
    if L<=mid:
        res+=query(L,R,p<<1,pl,mid)
    if R>mid:
        res+=query(L,R,p<<1|1,mid+1,pr)
    return res

n,m=map(int,input().split())
a=[0]+list(map(int,input().split()))
tag=[0]*(len(a)<<2)      #4倍大
tree=[0]*(len(a)<<2)
bulid(1,1,n)            #建树
for i in range(m):
    w=list(map(int,input().split()))
    if len(w)==3:       #区间查询:[L,R]的区间和
        q,L,R=w
        print(query(L,R,1,1,n))
    else:               #区间修改:把[L,R]的每个元素加上d
        q,L,R,d=w
        update(L,R,1,1,n,d)

1）build(）建树

2）update() 函数更新区间的值，把区间内所有元素的值加上 d。如果 tree[p] 这棵子树完全被包含在需要修改的区间 [L, R] 中，只需要对根 tree[p] 打上标记即可，不用修改 p 的子结点。

3）addtag() 打标记

4）update() 中的 push_down。update() 函数更新区间的值，把区间内所有元素的值加上 d。如果 tree[p] 这棵子树不能完全被包含在需要修改的区间 [L,R] 中，需要解决多次修改的冲突问题，用 push_down() 实现。

解决 tag 的冲突问题。把 tag 分别传给左右子树。

注意：tag 应该持续向下传递，直到能覆盖区间为止。但是 push_down() 只向下传递了一次。因为使用 push_down() 的 update() 是个递归函数，update() 会在递归时一层层地用 push_down() 来传递 tag。

5）query() 查询区间和。查询时用到 tag，也就是用 push_down() 来处理 tag。

【总结】

线段树是一个综合的知识点，它能锻炼以下能力：

1）计算理论：二分法、二叉树。

2）逻辑思维：懒惰标记技术 lazy-tag。

3）代码能力：递归。

请大量做线段树的练习，对线段树的掌握说明进入了算法竞赛的大门。而且线段树的扩展应用极多，可阅读资料了解。

【习题】

 以上，线段树的懒标记与应用

祝好