【数据结构】普通线段树与lazy标记

default111

于 2022-07-14 21:43:48 发布

阅读量355

点赞数

分类专栏：算法竞赛文章标签：数据结构大数据算法

本文链接：https://blog.csdn.net/default012/article/details/125793390

版权

算法竞赛专栏收录该内容

31 篇文章 3 订阅

订阅专栏

普通线段树用于维护区间的特殊信息（如和、最值、状态值等），lazy标记用于存储曾经进行过的修改，只有当需要查询时才实际更新区间信息
更好的阅读体验：我的博客

普通线段树

维护区间的特殊信息
本质：将区间不断分成左右两部分，每个节点用来表示一个区间，利用二叉树结构维护区间信息，所以父节点 $u$ 的子节点编号是 $2 u$ 和 $2 u + 1$ ，区间 $[L, R]$ 的左右区间是 $[L, mi d]$ 和 $(mi d, R]$
建树：递归建左右区间
维护/查询操作：当前区间为 $[L, R]$ ，须更新区间为 $[l, r]$ ，先求出 $L, R$ 的中间值 $mi d$ ，有三种情况：
- 若维护/查询区间是左区间 $[L, mi d]$ 的子区间，就以 $[l, r]$ 递归维护/查询左区间
- 若维护/查询区间是右区间 $(mi d, R]$ 的子区间，就以 $[l, r]$ 递归维护/查询右区间
- 若维护/查询区间横跨 $mi d$ ，则分别以 $[l, mi d], (mi d, r]$ 递归维护/查询左右区间
- 维护操作在上述操作结束后，进行pushdown 操作，即用子区间更新后的信息来更新当前区间的信息
一般开4倍空间，修改/查询时间复杂度 $O(\log N)$

lazy标记

对于系列更新操作，若不查询，即不需要使用更新后的值，那就没有必要更新，也不需要告诉子节点“需要更新”。
更新具有连续性，例如，先+1，再+2，可以在需要时直接+3，而不用分两步。
$\Rightarrow$ 把需要更新的内容记录在区间对应的节点上，等到需要的时候再更新，并将更新内容传递给子节点。

带lazy标记的普通线段树代码（节点信息以数组存储，没有写成结构体）：

int L[N << 2], R[N << 2];
ll sum[N << 2], lazy[N << 2];
inline int ls(int x) { return x << 1; }
inline int rs(int x) { return x << 1 | 1; }
inline void push_up(int rt) { sum[rt] = (sum[ls(rt)] + sum[rs(rt)]) % mod; }
inline void push_down(int rt)
{
    if (!lazy[rt])
        return;
    lazy[ls(rt)] += lazy[rt], lazy[rs(rt)] += lazy[rt];
    sum[ls(rt)] += lazy[rt] * (R[ls(rt)] - L[ls(rt)] + 1);
    sum[rs(rt)] += lazy[rt] * (R[rs(rt)] - L[rs(rt)] + 1);
    lazy[rt] = 0;
}
void build(int rt, int l, int r)
{
    L[rt] = l, R[rt] = r;
    if (l == r)
    {
        scanf("%lld", &sum[rt]);
        return;
    }
    int mid = (l + r) >> 1;
    build(ls(rt), l, mid);
    build(rs(rt), mid + 1, r);
    push_up(rt);
}
void update(int rt, int l, int r, ll val)
{
    if (L[rt] == l && R[rt] == r)
    {
        lazy[rt] += val;
        sum[rt] += val * (r - l + 1);
        return;
    }
    push_down(rt);
    int mid = (L[rt] + R[rt]) >> 1;
    if (r <= mid)
        add(ls(rt), l, r, val);
    else if (l > mid)
        add(rs(rt), l, r, val);
    else
        add(ls(rt), l, mid, val), add(rs(rt), mid + 1, r, val);
    push_up(rt);
}
ll query(int rt, int l, int r)
{
    if (L[rt] == l && R[rt] == r)
        return sum[rt];
    push_down(rt);
    int mid = (L[rt] + R[rt]) >> 1;
    if (r <= mid)
        return query(ls(rt), l, r);
    if (l > mid)
        return query(rs(rt), l, r);
    return query(ls(rt), l, mid) + query(rs(rt), mid + 1, r);
}