洛谷 P3372 【模板】线段树 1(延迟标记)

线段树，延迟标记
本题要点：
1、延迟标记：
当前的修改或者查询的区间范围是[l, r], 如果某个节点的区间 p 被 [l, r] 所覆盖（l <= tree[p].l && tree[p].r <= r）,
并且这个节点p ，在后续再也没有访问到，这个节点的子节点的相关信息可以不修改。 如果 p节点后续访问到了，只有在访问到
的时候，先更新 p 节点的所有子节点信息。 至于延迟标记，可以在 节点的结构体中，增加 add 标记。
查询到某节点 p， 根据 add 标记， 先 spread§, 再做后续操作。
2、 本题，区间 [l, r] 每个数加上 v， 并且查询任意区间 [l, r] 所有元素的和。 因此，在 线段树节点结构体维护的信息
区间范围 l, r; 区间总和 sum; 区间延迟标记 add .

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MaxN = 5e5 + 10;
int a[MaxN];
int n, m;

struct segTree
{
	int l, r;
	long long sum, add;
}tree[MaxN * 4];

void build(int p, int l, int r)
{
	tree[p].l = l, tree[p].r = r;
	if(l == r)
	{
		tree[p].sum = a[l];
		return;
	}
	int mid = (l + r) / 2;
	build(p * 2, l, mid);
	build(p * 2 + 1, mid + 1, r);
	tree[p].sum = tree[p * 2].sum + tree[p * 2 + 1].sum;
}

void spread(int p)	// 从节点p向下递归
{
	if(tree[p].add)
	{
		tree[p * 2].sum += tree[p].add * (tree[p * 2].r - tree[p * 2].l + 1);	
		tree[p * 2 + 1].sum += tree[p].add * (tree[p * 2 + 1].r - tree[p * 2 + 1].l + 1);
		tree[p * 2].add += tree[p].add;		//给左孩子打上延迟标记
		tree[p * 2 + 1].add += tree[p].add;	//给右孩子打上延迟标记
		tree[p].add = 0;					//清除p的标记
	}
}

void change(int p, int l, int r, int d)
{
	if(l <= tree[p].l && r >= tree[p].r)
	{
		tree[p].sum += (long long)d * (tree[p].r - tree[p].l + 1);	//更新节点信息
		tree[p].add += d;	//给节点打上延迟标记
		return;
	}
	spread(p);
	int mid = (tree[p].l + tree[p].r) / 2;
	if(l <= mid)
	{
		change(p * 2, l, r, d);	
	}
	if(r > mid)
	{
		change(p * 2 + 1, l, r, d);
	}
	tree[p].sum = tree[2 * p].sum + tree[2 * p + 1].sum;
}

long long ask(int p, int l, int r)
{
	if(l <= tree[p].l && r >= tree[p].r)
	{
		return tree[p].sum;
	}
	spread(p);
	int mid = (tree[p].l + tree[p].r) / 2;
	long long val = 0;
	if(l <= mid)
	{
		val += ask(p * 2, l, r);
	}
	if(r > mid)
	{
		val += ask(p * 2 + 1, l, r);
	}
	return val;
}

int main()
{
	int cmd, x, y, v;
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
	}
	build(1, 1, n);
	for(int i = 0; i < m; ++i)
	{
		scanf("%d", &cmd);
		if(1 == cmd)
		{
			scanf("%d%d%d", &x, &y, &v);	
			change(1, x, y, v);
		}else{
			scanf("%d%d", &x, &y);	
			printf("%lld\n", ask(1, x, y));
		}
	}
	return 0;
}

/*
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
*/

/*
11
8
20
*/