【模板】线段树2

题意简述

已知一个数列，你需要进行下面三种操作：
1.将某区间每一个数乘上x
2.将某区间每一个数加上x
3.求出某区间每一个数的和

代码

#include <cstdio>
using namespace std;
typedef long long ll;
ll n, m, opt, x, y, k, mod;
ll a[400010], la1[400010], la2[400010];
void push_up(ll x)
{
	a[x] = (a[x << 1] + a[x << 1 | 1]) % mod;
}
void push_down(ll x, ll len)
{
	if (la2[x] != 1)
	{
		(a[x << 1] *= la2[x]) %= mod;
		(a[x << 1 | 1] *= la2[x]) %= mod;
		(la1[x << 1] *= la2[x]) %= mod;
		(la1[x << 1 | 1] *= la2[x]) %= mod;
		(la2[x << 1] *= la2[x]) %= mod;
		(la2[x << 1 | 1] *= la2[x]) %= mod;
		la2[x] = 1;
	}
	if (la1[x])
	{
		(a[x << 1] += la1[x] * (len - (len >> 1)) % mod) %= mod;
		(a[x << 1 | 1] += la1[x] * (len >> 1) % mod) %= mod;
		(la1[x << 1] += la1[x]) %= mod;
		(la1[x << 1 | 1] += la1[x]) %= mod;
		la1[x] = 0;
	}
}
void build(ll x, ll l, ll r)
{
	la2[x] = 1;
	if (l == r)
	{
		scanf("%lld", &a[x]);
		return;
	}
	ll mid = l + r >> 1;
	build(x << 1, l, mid);
	build(x << 1 | 1, mid + 1, r);
	push_up(x);
}
void multiply(ll x, ll l, ll r, ll l1, ll r1)
{
	if(l1 <= l && r <= r1)
	{
		(a[x] *= k) %= mod;
		(la1[x] *= k) %= mod;
		(la2[x] *= k) %= mod;
		return;
	}
	push_down(x, r - l + 1);
	ll mid = l + r >> 1;
	if (l1 <= mid) multiply(x << 1, l, mid, l1, r1);
	if (r1 >  mid) multiply(x << 1 | 1, mid + 1, r, l1, r1);
	push_up(x);
}
void add(ll x, ll l, ll r, ll l1, ll r1)
{
	if (l1 <= l && r <= r1)
	{
		(a[x] += (r - l + 1) * k % mod) %= mod;
		(la1[x] += k) %= mod;
		return;
	}
	push_down(x, r - l + 1);
	ll mid = l + r >> 1;
	if (l1 <= mid) add(x << 1, l, mid, l1, r1);
	if (r1 >  mid) add(x << 1 | 1, mid + 1, r, l1, r1);
	push_up(x);
}
ll query(ll x, ll l, ll r, ll l1, ll r1, ll ans = 0)
{
	if (l1 <= l && r <= r1)	return a[x];
	push_down(x, r - l + 1);
	ll mid = l + r >> 1;
	if (l1 <= mid) (ans += query(x << 1, l, mid, l1, r1)) %= mod;
	if (r1 >  mid) (ans += query(x << 1 | 1, mid + 1, r, l1, r1)) %= mod;
	return ans;
}
int main()
{
	scanf("%lld%lld%lld", &n, &m, &mod);
	build(1, 1, n);
	for (register ll i = 1; i <= m; ++i)
	{
		scanf("%lld", &opt);
		if (opt == 1)
		{
			scanf("%lld%lld%lld", &x, &y, &k);
			multiply(1, 1, n, x, y);
		}
		if (opt == 2)
		{
			scanf("%lld%lld%lld", &x, &y, &k);
			add(1, 1, n, x, y);
		}
		if (opt == 3)
		{
			scanf("%lld%lld", &x, &y);
			printf("%lld
", query(1, 1, n, x, y));
		}
	}
}