题目链接:传送门
题目描述
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。 有长为N的数列,不妨设为a1,a2,…,aN 。有如下三种操作形式:
(1)把数列中的一段数全部乘一个值;
(2)把数列中的一段数全部加一个值;
(3)询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模P的值。
输入输出格式
第一行两个整数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,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
输入输出样例
输入样例#1:
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
输出样例#1:
2 35 8
说明
初始时数列为(1,2,3,4,5,6,7)。
经过第1次操作后,数列为(1,10,15,20,25,6,7)。
对第2次操作,和为10+15+20=45,模43的结果是2。
经过第3次操作后,数列为(1,10,24,29,34,15,16}
对第4次操作,和为1+10+24=35,模43的结果是35。
对第5次操作,和为29+34+15+16=94,模43的结果是8。
【题解】:
这个题目就是一个模板题。但是万事开头难,自己也错了不少次才AC的。
这个题目每一个节点不是以前的单纯记录Add[ ]即可。
还多了一个乘法的操作:
因为多了一个乘法的操作,所以必须增加一个mul[ ] 与之对应才行。
后来遇到问题了,又有乘法又有加法怎么弄呢????
这个问题真的弄得头都大了。
后来瞄了一眼题解才知道,原来只要一直满足先乘法后加法才可以弄。
如果当前位置有乘法,想转移到子节点,
那么pushdown操作必须先乘后加。
首先当前节点 mul = k,add = c
k(ax+b)=ak x + bk
然后 ak *x + b*k+c
如果到了叶子节点那么就是直接把对应的位置算出即可。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 100;
int n, m, L, R, Ins;
ll tree[N << 2], mul[N << 2], add[N << 2], a[N];
ll mod, c, tmp;
void Build(int No, int L, int R) {
mul[No] = 1;
int Mid = (L + R) >> 1;
if (L == R) {
tree[No] = a[L];
return;
}
Build(No << 1, L, Mid);
Build(No << 1 | 1, Mid + 1, R);
tree[No] = (tree[No << 1] + tree[No << 1 | 1]) % mod;
}
void Mul(int No, int L, int R, ll k) {
mul[No] = mul[No] * k % mod;
add[No] = add[No] * k % mod;
tree[No] = (tree[No] * k) % mod;
}
void Add(int No, int L, int R, ll k) {
add[No] = (add[No] + k) % mod;
tree[No] = (tree[No] + (R - L + 1) * k % mod) % mod;
}
void Push_down(int No, int L, int R) {
int Mid = (L + R) >> 1;
if (mul[No] == 1 && add[No] == 0)
return;
Mul(No << 1, L, Mid, mul[No]);
Mul(No << 1 | 1, Mid + 1, R, mul[No]);
Add(No << 1, L, Mid, add[No]);
Add(No << 1 | 1, Mid + 1, R, add[No]);
mul[No] = 1;
add[No] = 0;
}
void modify(int No, int L, int R, int x, int y, ll k, int f) {
if (x <= L && R <= y && f == 1) {
tree[No] = tree[No] * k % mod;
mul[No] = mul[No] * k % mod;
add[No] = add[No] * k % mod;
return;
}
if (x <= L && R <= y && f == 2) {
tree[No] = (tree[No] + (R - L + 1) * k) % mod;
add[No] = (add[No] + k) % mod;
return;
}
int Mid = (L + R) >> 1;
Push_down(No, L, R);
if (x <= Mid)
modify(No << 1, L, Mid, x, y, k, f);
if (Mid < y)
modify(No << 1 | 1, Mid + 1, R, x, y, k, f);
tree[No] = (tree[No << 1] + tree[No << 1 | 1]) % mod;
}
ll query(int No, int L, int R, int x, int y) {
if (x <= L && R <= y)
return tree[No];
int Mid = (L + R) >> 1;
Push_down(No, L, R);
ll res = 0;
if (x <= Mid)
res = (res + query(No << 1, L, Mid, x, y)) % mod;
if (Mid < y)
res = (res + query(No << 1 | 1, Mid + 1, R, x, y)) % mod;
return res % mod;
}
void check(int No, int L, int R) {
int Mid = (L + R) >> 1;
if (L == R) {
printf("(%d,%lld)\n", L, tree[No]);
return;
}
check(No << 1, L, Mid);
check(No << 1 | 1, Mid + 1, R);
}
int main() {
scanf("%d%lld", &n, &mod);
for (int i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
}
Build(1, 1, n);
scanf("%d", &m);
while (m--) {
scanf("%d%d%d", &Ins, &L, &R);
switch (Ins) {
case 1:
scanf("%lld", &c);
modify(1, 1, n, L, R, c, 1);
// check(1,1,n);
break;
case 2:
scanf("%lld", &c);
modify(1, 1, n, L, R, c, 2);
// check(1,1,n);
break;
case 3:
tmp = query(1, 1, n, L, R);
printf("%lld\n", tmp);
break;
}
}
return 0;
}