(线段树 + add懒标记 + mul懒标记)AcWing - 维护序列


原题链接:

AcWing - 维护序列

题目描述:

老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。
有长为 N 的数列,不妨设为 a1, a2, …, aN。
有如下三种操作形式:
1、把数列中的一段数全部乘一个值;
2、把数列中的一段数全部加一个值;
3、询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模 P 的值。
输入格式
第一行两个整数 N 和 P;
第二行含有 N 个非负整数,从左到右依次为 a1,a2,…,aN;
第三行有一个整数 M,表示操作总数;
从第四行开始每行描述一个操作,输入的操作有以下三种形式:
 操作 1:1 t g c,表示把所有满足 t≤i≤g 的 ai 改为 ai×c;
 操作 2:2 t g c,表示把所有满足 t≤i≤g 的 ai 改为 ai+c;
 操作 3:3 t g,询问所有满足 t≤i≤g 的 ai 的和模 P 的值。
同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。
输出格式
对每个操作 3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
数据范围
1≤N,M≤1e5,
1≤t≤g≤N,
0≤c,ai≤1e9,
1≤P≤1e9
输入样例:

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

思路:

  线段树中的值有l, r, sum,这是肯定的,然后还有懒标记add和mul,
那么根据懒标记更新sum时是先乘,还是先加呢,那就难点来了,怎样推出个公式呢:

对于x
若对懒标记的处理是先加再乘
    若此次操作为乘上一个数c
        可以表示为 (n + add) * mul * c 即 (n + X) * X 的形式
    若此次操作为加上一个数c
        (n + add) * mul + c 不能写成 (n + X ) * X的形式
        -> 无法更新新的懒标记

若对懒标记的处理是先乘再加
    若此次操作是加上一个数c
        可以表示为n * mul + add + c
        -> 此时新的add即为add + c
    若此次操作是乘上一个数c
        可以表示为n * mul * c + add * c
        -> 此时新的add即为add * c,新的mul即为mul * c 

-> 故先乘再加,以便更新懒标记

可以把乘和加的操作都看成 x * c + d
    -> 若是乘法,d为0
    -> 若是加法,c为1

若当前x的懒标记为add和mul
    -> 操作可以写成(x * mul + add) * c + d
    -> 即x * (mul * c) + (add * c + d)
    -> 新的mul为(mul * c),新的add为(add * c + d)

注意:乘的懒标记初始为1

作者:bigstone
链接:https://www.acwing.com/solution/content/43107/

  既然公式推出来了,那么我们就能愉快的写题了,和add懒标记的模板差不多.

#include
using namespace std;
#define endl '\n'
#define int long long
typedef pair PII;

const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10;
int n, p, m;
int w[N];
struct Node{
	int l, r;
	int sum, add, mul;
}tr[N << 2];

void pushup(int u){
	tr[u].sum = (tr[u << 1].sum + tr[u << 1 | 1].sum) % p;
}
void eval(Node &t, int mul, int add){
	t.sum = (t.sum * mul % p + (t.r - t.l + 1) * add) % p;
	t.mul = t.mul * mul % p;
	t.add = (t.add * mul + add) % p;
}
void pushdown(int u){
	eval(tr[u << 1], tr[u].mul, tr[u].add);
	eval(tr[u << 1 | 1], tr[u].mul, tr[u].add);
	tr[u].mul = 1, tr[u].add = 0;
}
void build(int u, int l, int r){
	tr[u].l = l, tr[u].r = r;
	tr[u].mul = 1;
	if(l == r) tr[u].sum = w[l];
	else{
		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, int add, int mul){
	if(tr[u].l >= l && tr[u].r <= r) eval(tr[u], mul, add);
	else{
		pushdown(u);
		int mid = tr[u].l + tr[u].r >> 1;
		if(l <= mid) modify(u << 1, l, r, add, mul);
		if(r > mid) modify(u << 1 | 1, l, r, add, mul);
		pushup(u);
	}
}
int query(int u, int l, int r){
	if(tr[u].l >= l && tr[u].r <= r) return tr[u].sum;
	pushdown(u);
	int mid = tr[u].l + tr[u].r >> 1;
	int 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;
}
signed main(void){
	ios::sync_with_stdio(false);
	cin.tie(0);
    cin >> n >> p;
    for(int i = 1; i <= n; i++)
    	cin >> w[i];
    build(1, 1, n);
    cin >> m;
    for(int i = 1; i <= m; i++){
    	int op, l, r, d;
    	cin >> op >> l >> r;
    	if(op == 1){
    		cin >> d;
    		modify(1, l, r, 0, d);
    	}
    	else if(op == 2){
    		cin >> d;
    		modify(1, l, r, d, 1);
    	}
    	else cout << query(1, l, r) << endl;
    }
	return 0;
}