洛谷 P3372 【模板】线段树 1

洛谷 P3372 【模板】线段树 1

传送门

思路

前几天学了线段树的我，今天又去做了一遍线段树【模板】\(1\)，发现自己打代码真的是漏洞百出啊，不过最后还是对了，所以来水一篇博客

首先，这道模板题的要求就是：

1.区间加
2.区间求和

这两个操作都属于线段树的基本操作

前置——宏定义

#define N 100011
#define lson rt << 1
#define rson rt << 1 | 1
#define int long long

由于我特别懒，不想写什么\(rt << 1\)之类的东西，所以直接宏定义就好了，还有，为了不改\(int\)，我直接把\(int\)宏定义为\(long \ long\)，省的麻烦（我真的是懒到家了）

1.定义

int sum[N << 2], lazy[N << 2], n, m;

众所周知，线段树一般要开四倍空间，不明白的可以自己画一颗线段树数一下，sum数组维护区间和，lazy是懒标记

2.pushup

inline void pushup(int rt) {
	sum[rt] = sum[lson] + sum[rson];
}

这是一个上传给父亲结点的信息的函数，当前节点的区间和就等于它左儿子和右儿子的区间和之和

2.建树

void build(int l, int r, int rt) {
	if(l == r) {
		sum[rt] = read();
		return;
	}
	int m = (l + r) >> 1;
	build(l, m, lson);
	build(m + 1, r, rson);
	pushup(rt);
}

上面是建树过程,\(rt\)表示当前节点，\(lson\)是左孩子，\(rson\) 是右孩子

3.标记下放

inline void pushdown(int l, int r, int rt) {
	if(!lazy[rt]) return;
	lazy[lson] += lazy[rt];
	lazy[rson] += lazy[rt];
	int m = (l + r) >> 1;
	sum[lson] += (m - l + 1) * lazy[rt];
	sum[rson] += (r - m) * lazy[rt];
	lazy[rt] = 0;
	return;
}

pushdown函数的实现：
1.用lazy存储这个懒标记。
2.递归到这个节点时，只更新这个节点的状态，并把当前的更改值累积到标记中。
什么时候才用到这个懒标记？
当需要递归这个节点的子节点时，标记下传给子节点。
3.下放操作：（三步）
当前节点的懒标记累积到子节点的懒标记中。
修改子节点状态。
父节点懒标记清0。

4.区间修改

void update(int L, int R, int c, int l, int r, int rt) {
	if(L <= l && r <= R) {
		sum[rt] += c * (r - l + 1);
		lazy[rt] += c;
		return;
	}
	pushdown(l, r, rt);
	int m = (l + r) >> 1;
	if(L <= m) update(L, R, c, l, m, lson);
	if(R > m) update(L, R, c, m + 1, r, rson);
	pushup(rt);
}

5.区间求和

int query(int L, int R, int l, int r, int rt) {
	if(L <= l && r <= R) return sum[rt];
	pushdown(l, r, rt);
	int m = (l + r) >> 1, ans = 0;
	if(L <= m) ans += query(L, R, l, m, lson);
	if(R > m) ans += query(L, R, m + 1, r, rson);
	return ans;
}

代码

下面上传高清无码无水印代码

#include <bits/stdc++.h>
#define N 100011
#define lson rt << 1
#define rson rt << 1 | 1
#define int long long
using namespace std;

int sum[N << 2], lazy[N << 2], n, m;

inline int read() {
	char c = getchar();
	int x = 0, f = 1;
	for( ; !isdigit(c); c = getchar()) if(c == '-') f = -1;
	for( ; isdigit(c); c = getchar()) x = (x << 3) + (x << 1) + (c ^ 48);
	return x * f;
}

inline void pushup(int rt) {
	sum[rt] = sum[lson] + sum[rson];
}

void build(int l, int r, int rt) {
	if(l == r) {
		sum[rt] = read();
		return;
	}
	int m = (l + r) >> 1;
	build(l, m, lson);
	build(m + 1, r, rson);
	pushup(rt);
}

inline void pushdown(int l, int r, int rt) {
	if(!lazy[rt]) return;
	lazy[lson] += lazy[rt];
	lazy[rson] += lazy[rt];
	int m = (l + r) >> 1;
	sum[lson] += (m - l + 1) * lazy[rt];
	sum[rson] += (r - m) * lazy[rt];
	lazy[rt] = 0;
	return;
}

void update(int L, int R, int c, int l, int r, int rt) {
	if(L <= l && r <= R) {
		sum[rt] += c * (r - l + 1);
		lazy[rt] += c;
		return;
	}
	pushdown(l, r, rt);
	int m = (l + r) >> 1;
	if(L <= m) update(L, R, c, l, m, lson);
	if(R > m) update(L, R, c, m + 1, r, rson);
	pushup(rt);
}

int query(int L, int R, int l, int r, int rt) {
	if(L <= l && r <= R) return sum[rt];
	pushdown(l, r, rt);
	int m = (l + r) >> 1, ans = 0;
	if(L <= m) ans += query(L, R, l, m, lson);
	if(R > m) ans += query(L, R, m + 1, r, rson);
	return ans;
}

signed main() {
	n = read(), m = read();
	build(1, n, 1);
	for(int i = 1; i <= m; i++) {
		int opt = read(), x = read(), y = read();
		if(opt == 1) {
			int k = read();
			update(x, y, k, 1, n, 1);
		}
		if(opt == 2) cout << query(x, y, 1, n, 1) << '\n';
	}
	return 0;
}