洛谷 P3373 【模板】线段树 2

$Description$

如题，已知一个数列，有下列三种操作：
操作1:将某区间每一个数乘上 $x$
操作2:将某区间每一个数加上 $x$
操作3:查询某区间每一个数的和

$Solution$

线段树模板
和【模板】线段树 1差不多
只不过操作2要处理一下
有两个下传标志$add,mul$

区间加

对于区间加，每个就直接加就好了，也就是$:$
$a[now].add=(a[now].add+k)modp$
$a[now].sum=(a[now].sum+k\times (a[now].r-a[now].l+1))modp$

区间乘

根据乘法分配律，$add,sum,mul$同时乘$k$
$a[now].sum=(a[now].sum\times k)modp$
$a[now].mul=(a[now].mul\times k)modp$
$a[now].add=(a[now].add\times k)modp$

下传

$sum$下传就是子孩子先乘一个$mul$然后加上$add,$因为$add$已经乘过所以直接加就好了
$a[son].sum=(a[son].sum\times a[now].mul+a[now].add\times (a[son].r-a[son].l+1))modp$
$mul$下传就直接乘
$a[son].mul=(a[son].mul\times a[now].mul)modp$
$add$下传和$sum$相似
$a[son].add=(a[son].add\times a[now].mul+a[now].add)modp$

$Acceptedcode$

#include<cstdio>
#include<iostream>
#define int long long
#define ls (now << 1)
#define rs (now << 1 | 1)
#define mid (a[now].l + a[now].r >> 1)

using namespace std;

const int N = 1e5 + 5;

struct Tree {
	int l, r, add, sum, mul;
}a[N<<2];

int n, q, k, mod;

inline void read(int &f) {
	f = 0; char c = getchar();
	while (!isdigit(c)) c = getchar();
	while (isdigit(c)) f = f * 10 + c - 48, c = getchar();
	return;
}

void build(int now, int l, int r) {
	a[now] = (Tree){l, r, 0, 0, 1};
	if (l == r) {
		read(a[now].sum);
		a[now].sum %= mod;
		return;
	}
	build(ls, l, mid);
	build(rs, mid + 1, r);
	a[now].sum = (a[ls].sum + a[rs].sum) % mod;
}

void pd(int now) {
	a[ls].sum = (a[ls].sum * a[now].mul + a[now].add * (a[ls].r - a[ls].l + 1)) % mod;
	a[rs].sum = (a[rs].sum * a[now].mul + a[now].add * (a[rs].r - a[rs].l + 1)) % mod;
	(a[ls].mul *= a[now].mul) %= mod;
	(a[rs].mul *= a[now].mul) %= mod;
	a[ls].add = (a[ls].add * a[now].mul + a[now].add) % mod;
	a[rs].add = (a[rs].add * a[now].mul + a[now].add) % mod;
	a[now].mul = 1, a[now].add = 0;
}

void mul(int now, int l, int r, int w) {
	if (a[now].l >= l && a[now].r <= r) {
		(a[now].sum *= w) %= mod;
		(a[now].mul *= w) %= mod;
		(a[now].add *= w) %= mod;
		return;
	}
	pd(now);
	if (l <= mid) mul(ls, l, r, w);
	if (r > mid) mul(rs, l, r, w);
	a[now].sum = (a[ls].sum + a[rs].sum) % mod;
	return;
}

void add(int now, int l, int r, int w) {
	if (a[now].l >= l && a[now].r <= r) {
		(a[now].add += w) %= mod;
		(a[now].sum += w * (a[now].r - a[now].l + 1)) %= mod;
		return;
	}
	pd(now);
	if (l <= mid) add(ls, l, r, w);
	if (r > mid) add(rs, l, r, w);
	a[now].sum = (a[ls].sum + a[rs].sum) % mod;
	return;
}

int ask(int now, int l, int r) {
	if (a[now].l >= l && a[now].r <= r)
		return a[now].sum;
	pd(now);
	int sum = 0;
	if (l <= mid) (sum += ask(ls, l, r)) %= mod;
	if (r > mid) (sum += ask(rs, l, r)) %= mod;
	return sum;
}

signed main() {
	read(n), read(q), read(mod);
	build(1, 1, n);
	while (q--) {
		int flag, cl, cr, k;
		read(flag), read(cl), read(cr);
		if (flag == 1) read(k), mul(1, cl, cr, k);
		else if (flag == 2) read(k), add(1, cl, cr, k);
		else printf("%lld\n", ask(1, cl, cr));
	}
}