【洛谷】P2023 维护序列

题目地址：

https://www.luogu.com.cn/problem/P2023

题目背景：
老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。

题目描述：
有一个长为$n$的数列 { a n } \{a_n\} {an​}，有如下三种操作形式：
1.格式1 t g c表示把所有满足$t\le i\le g$的$a_i$改为$a_i\times c$；
2.格式2 t g c表示把所有满足$t\le i\le g$的$a_i$改为$a_i+c$；
3.格式3 t g询问所有满足$t\le i\le g$的$a_i$的和，由于答案可能很大，你只需输出这个数模 $p$ 的值。

输入格式：
第一行两个整数$n$和$p$。
第二行含有$n$个非负整数，表示数列 { a i } \{a_i\} {ai​}。
第三行有一个整数 $m$，表示操作总数。
从第四行开始每行描述一个操作，同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

输出格式：
对每个操作$3$，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

数据范围：
测试数据规模如下表所示：

数据点编号	1	2	3	4	5	6	7	8	9,10
$n=$	$10$	$1000$	$1000$	$10000$	$60000$	$70000$	$80000$	$90000$	$100000$
$m=$	$10$	$1000$	$1000$	$10000$	$60000$	$70000$	$80000$	$90000$	$100000$

对于全部的测试点，保证$0\le p,a_i,c\le 10^9$，$1\le t\le g\le n$。

参考https://blog.csdn.net/qq_46105170/article/details/121469108。代码如下：

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int n, m, P;
int a[N];
struct Node {
  int l, r;
  long sum;
  long mul, add;
} tr[N << 2];

void pushup(int u) {
  tr[u].sum = (tr[u << 1].sum + tr[u << 1 | 1].sum) % P;
}

void update(Node &u, long mul, long add) {
  u.sum = (u.sum * mul + add * (u.r - u.l + 1)) % P;
  u.mul = u.mul * mul % P;
  u.add = (u.add * mul + add) % P;
}

void pushdown(int u) {
  auto &mul = tr[u].mul, &add = tr[u].add;
  update(tr[u << 1], mul, add), update(tr[u << 1 | 1], mul, add);
  mul = 1;
  add = 0;
}

void build(int u, int l, int r) {
  tr[u] = {l, r, 0, 1, 0};
  if (l == r) {
	tr[u].sum = a[l] % P;
	return;
  }
  int mid = l + r >> 1;
  build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
  pushup(u);
}

void modify(int u, int l, int r, long mul, long add) {
  if (l <= tr[u].l && tr[u].r <= r) {
	update(tr[u], mul, add);
	return;
  }

  pushdown(u);
  int mid = tr[u].l + tr[u].r >> 1;
  if (l <= mid) modify(u << 1, l, r, mul, add);
  if (r > mid) modify(u << 1 | 1, l, r, mul, add);
  pushup(u);
}

long query(int u, int l, int r) {
  if (l <= tr[u].l && tr[u].r <= r) return tr[u].sum;
  pushdown(u);
  int mid = tr[u].l + tr[u].r >> 1;
  long res = 0;
  if (l <= mid) res = query(u << 1, l, r);
  if (r > mid) res = (res + query(u << 1 | 1, l, r)) % P;
  return res;
}

int main() {
  scanf("%d%d", &n, &P);
  for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
  build(1, 1, n);
  scanf("%d", &m);
  while (m--) {
	int op, l, r;
	long c;
	scanf("%d%d%d", &op, &l, &r);
	if (op < 3) scanf("%ld", &c);
	if (op == 1) modify(1, l, r, c, 0);
	else if (op == 2) modify(1, l, r, 1, c);
	else printf("%ld\n", query(1, l, r));
  }
}

每次操作时间复杂度$O(\log n)$，空间$O(n)$。