维护序列

维护序列

代码

#include<iostream>
#define MAXN 100005
#define ll long long
using namespace std;
int n, m, mod,k;
int a[MAXN];
struct Tree 
{
	ll sum, add, mul;
	int l, r;
}s[MAXN<<2];
void update(int pos) 
{
	s[pos].sum = (s[pos << 1].sum + s[pos << 1 | 1].sum) % mod;
	return;
}
void pushdown(int pos) 
{
	s[pos << 1].sum = (s[pos << 1].sum * s[pos].mul + s[pos].add * (s[pos << 1].r - s[pos << 1].l + 1)) % mod;
	s[pos << 1 | 1].sum = (s[pos << 1 | 1].sum * s[pos].mul + s[pos].add * (s[pos << 1 | 1].r - s[pos << 1 | 1].l + 1)) % mod;

	s[pos << 1].mul = (s[pos << 1].mul * s[pos].mul) % mod;
	s[pos << 1 | 1].mul = (s[pos << 1 | 1].mul * s[pos].mul) % mod;

	s[pos << 1].add = (s[pos << 1].add * s[pos].mul + s[pos].add) % mod;
	s[pos << 1 | 1].add = (s[pos << 1 | 1].add * s[pos].mul + s[pos].add) % mod;

	s[pos].add = 0;
	s[pos].mul = 1;
	return;
}
void build(int pos, int l, int r)
{ 
	s[pos].l = l;
	s[pos].r = r;
	s[pos].mul = 1;

	if (l == r) {
		s[pos].sum = a[l] % mod;
		return;
	}

	int mid = (l + r) >> 1;
	build(pos << 1, l, mid);
	build(pos << 1 | 1, mid + 1, r);
	update(pos);
	return;
}

void Mul(int pos, int x, int y, int k) 
{
	if (x <= s[pos].l && s[pos].r <= y) 
	{
		s[pos].add = (s[pos].add * k) % mod;
		s[pos].mul = (s[pos].mul * k) % mod;
		s[pos].sum = (s[pos].sum * k) % mod;
		return;
	}

	pushdown(pos);
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) Mul(pos << 1, x, y, k);
	if (y > mid) Mul(pos << 1 | 1, x, y, k);
	update(pos);
	return;
}

void Add(int pos, int x, int y, int k) 
{ 
	if (x <= s[pos].l && s[pos].r <= y)
	{
		s[pos].add = (s[pos].add + k) % mod;
		s[pos].sum = (s[pos].sum + k * (s[pos].r - s[pos].l + 1)) % mod;
		return;
	}
	pushdown(pos);
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) Add(pos << 1, x, y, k);
	if (y > mid) Add(pos << 1 | 1, x, y, k);
	update(pos);
	return;
}

ll query(int pos, int x, int y)
{
	if (x <= s[pos].l && s[pos].r <= y) 
	{
		return s[pos].sum;
	}
	pushdown(pos);
	ll res = 0;
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) res = (res + query(pos << 1, x, y)) % mod;
	if (y > mid) res = (res + query(pos << 1 | 1, x, y)) % mod;
	return res;
}
int main() 
{
	cin >> n >> mod;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	build(1, 1, n);
	cin >> m;
	while(m--)
	{
		int a, x, y;
		cin >> a >> x >> y;
		if (a == 1) 
		{
			cin >> k;
			Mul(1, x, y, k);
		}
		if (a == 2) 
		{
			cin >> k;
			Add(1, x, y, k);
		}
		if (a == 3)printf("%lld\n",query(1, x, y));
	}
	return 0;
}