bzoj1798: [Ahoi2009]Seq 维护序列seq（线段树）
题目描述：给定n个数的序列，有三种操作。
1、将一段区间乘上c
2、将一段区间加上c
3、求一段区间的和
输入格式：第一行两个整数第一行两个整数N和P(1≤P≤1000000000）。第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。第三行有一个整数M，表示操作总数。从第四行开始每行描述一个操作，输入的操作有以下三种形式： 操作1：“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c(1≤t≤g≤N,0≤c≤1000000000)。 操作2：“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。 操作3：“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。
输出格式：对每个操作3，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。
输入样例：
7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7
输出样例：
2
35
8
解析：线段树的模板题。关键在于如何处理线段树的乘和加。
可以将每一个数变为\(a \times b + c\) 的形式，在每一次乘的修改时，直接将和乘c，并将加和乘的标记都乘c即可。而加的修改则只需要将加的标记加上c即可。
在标记下传时，将该点子树的乘标记乘上该点的乘标记；将子树的加标记先乘上该点的乘标记，再加上该点的加标记即可。
代码如下：
#include<cstdio>
#define lc o << 1
#define rc o << 1 | 1
using namespace std;
const int maxn = 1e5 + 5;
int n, m, p, a[maxn], sum[maxn << 2], bj1[maxn << 2], bj2[maxn << 2]; //bj1:乘标记，bj2:加标记
int read(void) {
char c; while (c = getchar(), c < '0' || c >'9'); int x = c - '0';
while (c = getchar(), c >= '0' && c <= '9') x = x * 10 + c - '0'; return x;
}
void build(int o, int l, int r) { //建树
bj1[o] = 1; //乘标记的初始值为1
if (l == r) {
sum[o] = a[l] % p;
return;
}
int mid = l + r >> 1;
build(lc, l, mid); build(rc, mid + 1, r);
sum[o] = (sum[lc] + sum[rc]) % p;
}
void pushdown(int o, int l, int r) { //标记下传
int mid = l + r >> 1;
sum[lc] = 1ll * sum[lc] * bj1[o] % p; sum[lc] += 1ll * bj2[o] * (mid - l + 1) % p; sum[lc] %= p;
sum[rc] = 1ll * sum[rc] * bj1[o] % p; sum[rc] += 1ll * bj2[o] * (r - mid) % p; sum[rc] %= p;
bj1[lc] = 1ll * bj1[lc] * bj1[o] % p;
bj1[rc] = 1ll * bj1[rc] * bj1[o] % p;
bj2[lc] = 1ll * bj2[lc] * bj1[o] % p; bj2[lc] = (bj2[lc] + bj2[o]) % p;
bj2[rc] = 1ll * bj2[rc] * bj1[o] % p; bj2[rc] = (bj2[rc] + bj2[o]) % p;
bj1[o] = 1; bj2[o] = 0;
}
void modify1(int o, int l, int r, int ql, int qr,int c) {
if (ql <= l && qr >= r) { //两个标记都要乘上c
sum[o] = 1ll * sum[o] * c % p;
bj1[o] = 1ll * bj1[o] * c % p;
bj2[o] = 1ll * bj2[o] * c % p;
return;
}
int mid = l + r >> 1;
if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
if (ql <= mid) modify1(lc, l, mid, ql, qr, c);
if (qr > mid) modify1(rc, mid + 1, r, ql, qr, c);
sum[o] = (sum[lc] + sum[rc]) % p;
}
void modify2(int o, int l, int r, int ql, int qr, int c) {
if (ql <= l && qr >= r) { //加标记直接加上c
sum[o] += 1ll * c * (r - l + 1) % p; sum[o] %= p;
bj2[o] = (bj2[o] + c) % p;
return;
}
int mid = l + r >> 1;
if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
if (ql <= mid) modify2(lc, l, mid, ql, qr, c);
if (qr > mid) modify2(rc, mid + 1, r, ql, qr, c);
sum[o] = (sum[lc] + sum[rc]) % p;
}
int query(int o, int l, int r, int ql, int qr) {
if (ql <= l && qr >= r) return sum[o];
int mid = l + r >> 1, ans = 0;
if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
if (ql <= mid) ans += query(lc, l, mid, ql, qr); ans %= p;
if (qr > mid) ans += query(rc, mid + 1, r, ql, qr); ans %= p;
return ans;
}
int main() {
n = read(); p = read();
for (int i = 1; i <= n; ++ i) a[i] = read();
build(1, 1, n);
m = read();
while (m --) {
int opt = read();
if (opt == 1) {
int x = read(), y = read(), c = read();
modify1(1, 1, n, x, y, c);
}
else if (opt == 2) {
int x = read(), y = read(), c = read();
modify2(1, 1, n, x, y, c);
}
else {
int x = read(), y = read();
printf("%d\n", query(1, 1, n, x, y));
}
}
return 0;
}
