题目地址:
https://www.luogu.com.cn/problem/P2023
题目背景:
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。
题目描述:
有一个长为
n
n
n的数列
{
a
n
}
\{a_n\}
{an},有如下三种操作形式:
1.格式1 t g c
表示把所有满足
t
≤
i
≤
g
t\le i\le g
t≤i≤g的
a
i
a_i
ai改为
a
i
×
c
a_i\times c
ai×c;
2.格式2 t g c
表示把所有满足
t
≤
i
≤
g
t\le i\le g
t≤i≤g的
a
i
a_i
ai改为
a
i
+
c
a_i+c
ai+c;
3.格式3 t g
询问所有满足
t
≤
i
≤
g
t\le i\le g
t≤i≤g的
a
i
a_i
ai的和,由于答案可能很大,你只需输出这个数模
p
p
p 的值。
输入格式:
第一行两个整数
n
n
n和
p
p
p。
第二行含有
n
n
n个非负整数,表示数列
{
a
i
}
\{a_i\}
{ai}。
第三行有一个整数
m
m
m,表示操作总数。
从第四行开始每行描述一个操作,同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。
输出格式:
对每个操作
3
3
3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
数据范围:
测试数据规模如下表所示:
数据点编号 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9,10 |
---|---|---|---|---|---|---|---|---|---|
n = n= n= | 10 10 10 | 1000 1000 1000 | 1000 1000 1000 | 10000 10000 10000 | 60000 60000 60000 | 70000 70000 70000 | 80000 80000 80000 | 90000 90000 90000 | 100000 100000 100000 |
m = m= m= | 10 10 10 | 1000 1000 1000 | 1000 1000 1000 | 10000 10000 10000 | 60000 60000 60000 | 70000 70000 70000 | 80000 80000 80000 | 90000 90000 90000 | 100000 100000 100000 |
对于全部的测试点,保证 0 ≤ p , a i , c ≤ 1 0 9 0 \leq p, a_i, c \leq 10^9 0≤p,ai,c≤109, 1 ≤ t ≤ g ≤ n 1 \leq t \leq g \leq n 1≤t≤g≤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(\log n) O(logn),空间 O ( n ) O(n) O(n)。