P3372 【模板】线段树 1---线段树的区间修改

线段树的区间修改

线段树的单点修改很简单，很快，只用$O(lo{g}_{2}n)$
但用单点修改来实现区间修改就很费力了， 要$O((r-l+1)lo{g}_{2}n)$
所以我们不能这么做
要用到lazy标记

lazy标记

可以想象成发压岁钱，太爷爷要给孙子发压岁钱。
肯定要先把压岁钱发给爷爷，而爷爷这么想：“反正迟早要发下去的，先放在我这里”
过了几天，太爷爷来检查了，爷爷就发了下去，给了爸爸。
压岁钱发给爸爸，而爸爸也这么想：“反正迟早要发下去的，先放在我这里”
过了几天，太爷爷来检查了，爸爸就发了下去，给了儿子。
结束。

例题：P3372 【模板】线段树 1

#include <bits/stdc++.h>
#define int long long
#define ls id * 2
#define rs id * 2 + 1
using namespace std;
const int inf = 0x3f3f3f3f, N = 1e5 + 5, MOD = 1e9 + 7;
const int INF = 0x3f3f3f3f3f3f3f3f;
int lazy[N * 4], sumv[N * 4], a[N];
void build(int id, int l, int r)
{
	if (l == r)
	{
		sumv[id] = a[l];
		return;
	}
	int mid = (l + r) / 2;
	build(ls, l, mid);
	build(rs, mid + 1, r);
	sumv[id] = sumv[ls] + sumv[rs];
}

// x要发给他的左儿子和右儿子
void down(int id, int l, int r)
{
	if (lazy[id])
	{
		int mid = (l + r) / 2;
		sumv[ls] += lazy[id] * (mid - l + 1);
		lazy[ls] += lazy[id];
		sumv[rs] += lazy[id] * (r - mid);
		lazy[rs] += lazy[id];
		lazy[id] = 0;
	}
}  

int query(int id, int l, int r, int x, int y)
{
	if (x <= l && r <= y)
		return sumv[id];
	down(id, l, r);
	int mid = (l + r) / 2;
	int ans = 0;
	if (x <= mid)
		ans = query(ls, l, mid, x, y);
	if (y > mid)
		ans += query(rs, mid + 1, r, x, y);
	return ans;
}

void change(int id, int l, int r, int x, int y, int u)
{
	if (x <= l && r <= y)
	{
		sumv[id] += (r - l + 1) * u;
		// 放在手上
		lazy[id] += u;
		return;
	}
	down(id, l, r);
	int mid = (l + r) / 2;
	if (x <= mid)
		change(ls, l, mid, x, y, u);
	if (y > mid)
		change(rs, mid + 1, r, x, y, u);
	sumv[id] = sumv[ls] + sumv[rs];
}

signed main()
{
	int n, q;
	scanf("%lld%lld", &n, &q);
	for (int i = 1; i <= n; i++)
		scanf("%lld", &a[i]);
	build(1, 1, n);
	while (q--)
	{
		int op, x, y, z;
		scanf("%lld%lld%lld", &op, &x, &y);
		if (op == 1)
		{
			scanf("%lld", &z);
			change(1, 1, n, x, y, z);
		}else
		{
			printf("%lld\n", query(1, 1, n, x, y));
		}
	}
	return 0;
}