普通线段树用于维护区间的特殊信息(如和、最值、状态值等),lazy标记用于存储曾经进行过的修改,只有当需要查询时才实际更新区间信息
更好的阅读体验:我的博客
普通线段树
- 维护区间的特殊信息
- 本质:将区间不断分成左右两部分,每个节点用来表示一个区间,利用二叉树结构维护区间信息,所以父节点 u u u 的子节点编号是 2 u 2u 2u 和 2 u + 1 2u+1 2u+1 ,区间 [ L , R ] [L,R] [L,R] 的左右区间是 [ L , m i d ] [L,mid] [L,mid] 和 ( m i d , R ] (mid,R] (mid,R]
- 建树:递归建左右区间
- 维护/查询操作:当前区间为
[
L
,
R
]
[L,R]
[L,R] ,须更新区间为
[
l
,
r
]
[l,r]
[l,r] ,先求出
L
,
R
L,R
L,R 的中间值
m
i
d
mid
mid ,有三种情况:
- 若维护/查询区间是左区间 [ L , m i d ] [L,mid] [L,mid] 的子区间,就以 [ l , r ] [l,r] [l,r] 递归维护/查询左区间
- 若维护/查询区间是右区间 ( m i d , R ] (mid,R] (mid,R] 的子区间,就以 [ l , r ] [l,r] [l,r] 递归维护/查询右区间
- 若维护/查询区间横跨 m i d mid mid ,则分别以 [ l , m i d ] , ( m i d , r ] [l,mid],(mid,r] [l,mid],(mid,r] 递归维护/查询左右区间
- 维护操作在上述操作结束后,进行
pushdown
操作,即用子区间更新后的信息来更新当前区间的信息
- 一般开4倍空间,修改/查询时间复杂度 O ( log N ) O(\log N) O(logN)
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);
}
- 乘法和加法同时维护pushdown的修改:
- 两个懒标记
- 孩子sum=孩子sum*父亲乘法标记+区间长度*父亲加法标记
- 孩子乘法标记*=父亲乘法标记
- 孩子加法标记=孩子加法标记*父亲乘法标记+父亲加法标记
- 父亲加法标记=0,乘法标记=1