线段树 +懒标记 + P3372 【模板】线段树 1

题目描述
如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入格式
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式
输出包含若干行整数，即为所有操作2的结果。

输入输出样例
输入 #1复制
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
输出 #1复制
11
8
20
说明/提示
时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_，保证在int64/long long数据范围内）

样例说明：

在这里插入图片描述
代码

#include<bits/stdc++.h>
using namespace std;
#define Mid ((l + r)>>1)
#define lson node<<1, l, Mid
#define rson node<<1|1, Mid + 1, r
typedef long long ll;
ll arr[100005];
ll tree[100005*4];
ll lazy[100005*4];
void push_down(ll node, ll len)
{
    tree[node<<1] += (len-(len>>1))*lazy[node];
    tree[node<<1|1] += (len>>1)*lazy[node];
    lazy[node<<1] += lazy[node];
    lazy[node<<1|1] += lazy[node];
    lazy[node] = 0;
}
void update_tree(ll node, ll l, ll r, ll x, ll y, ll k)
{
	if(x <= l && r <= y)
	{
		tree[node] += (r - l + 1) * k;
		lazy[node] += k;
		return;
	}	
	else
	{
		if(lazy[node])
		{
			push_down(node, r - l + 1);
		}
		if(x <= Mid)
		{
			update_tree(lson, x, y, k);
		}
		if(y > Mid)
		{
			update_tree(rson, x, y, k);
		}
		tree[node] = tree[node << 1] + tree[node <<1|1];
	}
}
ll query_tree(ll node, ll l, ll r, ll x, ll y)
{
	ll lsum = 0;
	ll rsum = 0;
	if(x <= l && r <= y)
	{
		return tree[node];
	}
	if(lazy[node])
	{
		push_down(node, r - l + 1);
	}
	if(x <= Mid)
	{
		lsum =  query_tree(lson, x, y);
	}
	if(y > Mid)
	{
		rsum =  query_tree(rson, x, y);
	}
	return lsum + rsum;
	
}
void build_tree(ll node, ll l, ll r)
{
	if(l == r)
	{
		tree[node] = arr[l];
	}
	else
	{
		build_tree(lson);
		build_tree(rson);
		tree[node] = tree[node<<1] + tree[node << 1|1];
	}
}
int main()
{
	ll n, m;
	scanf("%lld %lld", &n, &m);
	getchar();
	for(ll i = 1; i <= n; i++)
	{
		scanf("%lld", &arr[i]);
	}
	build_tree(1, 1, n);
	
	for(ll i = 1; i <= m; i++)
	{
		ll x;
		scanf("%lld", &x);
		if(x == 1)
		{
			ll L, R, K;
			scanf("%lld %lld %lld", &L, &R, &K);
			update_tree(1, 1, n, L, R, K);
		}
		if(x == 2)
		{
			ll L, R;
			scanf("%lld %lld", &L, &R);
			ll sum = query_tree(1, 1, n, L, R);
			printf("%lld\n", sum);
		}
	}
	return 0;
}