P3372 【模板】线段树 1 （延迟标记）

区间修改，区间和查询。懒标记。

存个模板。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
struct T {
	int l, r;
	ll sum, add;//add为懒标记
	#define l(x) t[x].l
	#define r(x) t[x].r
	#define sum(x) t[x].sum
	#define add(x) t[x].add 
}t[N << 2];
int a[N], n, m;
void pushup(int p) {
	sum(p) = sum(p << 1) + sum(p << 1 | 1);
}
void build(int p, int l, int r) {
	l(p) = l, r(p) = r;
	if (l == r) {
		sum(p) = a[l];
		return ;
	}
	int mid = (l + r) >> 1;
	build(p << 1, l, mid);
	build(p << 1 | 1, mid + 1, r);
	pushup(p);
}
void pushdown(int p) {
	if (add(p)) {
		sum(p << 1) += add(p) * (r(p << 1) - l(p << 1) + 1);
		sum(p << 1 | 1) += add(p) * (r(p << 1 | 1) - l(p << 1 | 1) + 1);
		add(p << 1) += add(p);
		add(p << 1 | 1) += add(p);
		add(p) = 0;//清除标记*** 
	}
}
void modify(int p, int l, int r, int val) {
	if (l <= l(p) && r >= r(p)) {
		sum(p) += (ll) (r(p) - l(p) + 1) * val;
		add(p) += val;
		return ;
	}
	pushdown(p);//标准的正确写法 
	int mid = (l(p) + r(p)) >> 1;
	if (l <= mid) modify(p << 1, l, r, val);
	if (r > mid) modify(p << 1 | 1, l, r, val);
	pushup(p);
}
ll query(int p, int l, int r) {
	if (l <= l(p) && r >= r(p)) return sum(p);
	pushdown(p);
	int mid = (l(p) + r(p)) >> 1;
	ll res = 0;
	if (l <= mid) res += query(p << 1, l, r);
	if (r > mid) res += query(p << 1 | 1, l, r);
	return res;
}
int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i ++) cin >> a[i];
    build(1, 1, n);
    while (m --) {
    	int op, x, y, k;
    	cin >> op >> x >> y;
    	if (op == 1) {
    		cin >> k;
    		modify(1, x, y, k);
		}
		else {
			cout << query(1, x, y) << endl;
		}
	}
	return 0;
}