确定线段树维护的信息
根据题意设计节点所需要维护的信息,可以设计多个其目的是维护最终信息,但每个信息可简单方便的求出
如求区间最大连续子段和时如何设计维护的线段树:
struct node {
ll l, r;
ll totmx;//最大连续子段和
ll lmax;//最大前缀和
ll rmax;//最大后缀和
ll sum;//区间和
}tree[N << 2];
设计push_up()函数
在左右儿子信息正确的情况下如何更新父节点信息
如求区间最大连续子段和时如何去维护父节点信息:
void push_up(ll root) {
tree[root].sum = tree[root << 1].sum + tree[root << 1 | 1].sum;
tree[root].lmax = max(tree[root << 1].sum + tree[root << 1 | 1].lmax, tree[root << 1].lmax);
tree[root].rmax = max(tree[root << 1 | 1].sum + tree[root << 1].rmax, tree[root << 1 | 1].rmax);
tree[root].totmx = max(tree[root << 1].rmax + tree[root << 1 | 1].lmax, max(tree[root << 1].totmx, tree[root << 1 | 1].totmx));
}
对于建树中的push_up函数来说:
对于修改中的push_up函数来说:
建立线段树
递归的建立线段树结构,递归尾是节点的左区间=节点的右区间(叶节点),叶节点的每个信息维护好后利用push_up函数确定好树中每个节点维护的信息。
建立树一般来说是模板:
void build(int root, int l, int r) {
if (l == r)tree[root] = { /*顺序填入维护的值*/ };
else {
tree[root] = {/*初值*/ };//要赋初值
int mid = l + r >> 1;
build(root << 1, l, mid);
build(root << 1 | 1, mid + 1, r);
push_up(root);
}
}
push_down()函数
在父节点信息正确的情况下如何利用父节点的懒标记信息更新左右子节点
考虑到叶子节点的懒标记可能出现积累的情况,所以节点的各信息都利用原有的信息和父节点的信息更新
同时清除push_down后父节点的懒标记信息
对于查询操作中的push_dwon来说:
对于修改操作中的push_dowm来说:
修改函数
若节点的区间在所要修改的区间范围内直接修改区间各信息的值,有懒标记更新懒标记
若节点区间不在所要修改的区间范围内(从根节点开始即最大的区间范围开始向子节点即小区间递归修改)如果不在区间先push_down确保子节点信息正确后,判断左右字节点区间是否在修改范围内有则修改,最后递归回来后push_up保证父节点信息正确
如单点修改可改变为区间修改的修改函数:
void modify(int root, int l, int r) {
if (tree[root].l == tree[root].r) {
if (tree[root].flag) tree[root].sum = (tree[root].sum * 2) % MOD;
else {
tree[root].sum = tree[root].sum + lowbit(tree[root].sum);
tree[root].flag = check(tree[root].sum);
}
return;
}
if (tree[root].l >= l && tree[root].r <= r) {
if (tree[root].flag) {
tree[root].cnt = (tree[root].cnt * 2) % MOD;
tree[root].sum = (tree[root].sum * 2) % MOD;
return;
}
}
push_down(root);
int mid = tree[root].l + tree[root].r >> 1;
if (l <= mid)modify(root << 1, l, r);
if (r > mid)modify(root << 1 | 1, l, r);
push_up(root);
}
查询函数
若节点的区间在所要查询的区间范围内,直接返回节点的维护好的信息
若节点区间不在所要查询的区间范围内(从根节点开始即最大的区间范围开始向子节点即小区间递归查询)如果不在区间先push_down确保子节点信息正确后,判断左右字节点区间是否在查询范围内有则查询
查询函数一般来说是模板:
/*返回值类型*/ query(int root, int l, int r) {
if (tree[root].l >= l && tree[root].r <= r) return /*要求的值如tree[root].sum*/;
push_down(root);
int mid = tree[root].l + tree[root].r >> 1;
if (r <= mid)return query(root << 1, l, r);
else if (l > mid)return query(root << 1 | 1, l, r);
else {
/*返回值类型*/ left = query(root << 1, l, r);
/*返回值类型*/ right = query(root << 1 | 1, l, r);
/*返回值类型*/ ans;
//push_up
ans = left /*操作*/ right;
return ans;
}
}