【模板】线段树 2

题目链接

题目描述

如题，已知一个数列，你需要进行下面三种操作：

• 将某区间每一个数乘上 x
• 将某区间每一个数加上 x
• 求出某区间每一个数的和

输入格式

第一行包含三个整数 n,m,p，分别表示该数列数字的个数、操作的总个数和模数。

第二行包含 n 个用空格分隔的整数，其中第 i 个数字表示数列第 i 项的初始值。

接下来 m 行每行包含若干个整数，表示一个操作，具体如下：

操作 1： 格式：1 x y k 含义：将区间 [x,y] 内每个数乘上 k

操作 2： 格式：2 x y k 含义：将区间 [x,y] 内每个数加上 kk

操作 3： 格式：3 x y 含义：输出区间 [x,y] 内每个数的和对 p 取模所得的结果

输出格式

输出包含若干行整数，即为所有操作 3 的结果。

输入输出样例

输入

5 5 38
1 5 4 2 3
2 1 4 1
3 2 5
1 2 4 2
2 3 5 5
3 1 4

输出

17
2

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll f[N];
int n, m, p;

struct node{
	int l, r;
	ll sum, add, mul;
}a[N*4];

void update(int k)	// 更新 k 的sum 
{
	a[k].sum = (a[k*2].sum + a[k*2+1].sum) % p;
}

void pushdown(int k)	// 将点 k 的懒惰标记下传 
{
	a[k*2].sum = (a[k*2].sum * a[k].mul + a[k].add * (a[k*2].r - a[k*2].l + 1)) % p;
	a[k*2+1].sum = (a[k*2+1].sum * a[k].mul + a[k].add * (a[k*2+1].r - a[k*2+1].l + 1)) % p;
	
	a[k*2].mul = (a[k*2].mul * a[k].mul) % p;
	a[k*2+1].mul = (a[k*2+1].mul * a[k].mul) % p;
	
	a[k*2].add = (a[k*2].add * a[k].mul + a[k].add) % p;
	a[k*2+1].add = (a[k*2+1].add * a[k].mul + a[k].add) % p;
	
	a[k].add = 0;
	a[k].mul = 1;
}

void build(int k, int l, int r)
{
	a[k].l = l;
	a[k].r = r;
	a[k].mul = 1;
	if(l == r){		// 递归到叶节点 
		a[k].sum = f[l];
		return ;
	}
	int mid = (l + r) / 2;
	build(k * 2, l, mid);
	build(k * 2 + 1, mid + 1, r);
	update(k);
}

void changeSegment1(int k, int l, int r, int x)	// 区间修改 
{
	if(a[k].l == l && a[k].r == r){
		a[k].add = (a[k].add * x) % p;
		a[k].mul = (a[k].mul * x) % p;
		a[k].sum = (a[k].sum * x) % p;	// 更新该区间的sum 		
		return ;
	}
	pushdown(k);
	int mid = (a[k].l + a[k].r) / 2;
	if(r <= mid) changeSegment1(k * 2, l, r, x);		// 被修改的区间在左区间 
	else if(l > mid) changeSegment1(k * 2 + 1, l, r, x);		// 被修改的区间在右区间 
	else{	// 把修改区间分解成两块，分别往左右区间递归 
		changeSegment1(k * 2, l, mid, x);
		changeSegment1(k * 2 + 1, mid + 1, r, x);
	}
	update(k);
}

void changeSegment2(int k, int l, int r, int x)	// 区间修改 
{
	if(a[k].l == l && a[k].r == r){
		a[k].add = (a[k].add + x) % p;
		a[k].sum = (a[k].sum + (r - l + 1) * x) % p;	// 更新该区间的sum 	
		return ;
	}
	pushdown(k);
	int mid = (a[k].l + a[k].r) / 2;
	if(r <= mid) changeSegment2(k * 2, l, r, x);		// 被修改的区间在左区间 
	else if(l > mid) changeSegment2(k * 2 + 1, l, r, x);		// 被修改的区间在右区间 
	else{	// 把修改区间分解成两块，分别往左右区间递归 
		changeSegment2(k * 2, l, mid, x);
		changeSegment2(k * 2 + 1, mid + 1, r, x);
	}
	update(k);
}

ll query(int k, int l, int r)	// 区间查询
{
	if(a[k].l == l && a[k].r == r) return a[k].sum % p;
	pushdown(k);
	int mid = (a[k].l + a[k].r) / 2;
	if(r <= mid) return query(k * 2, l, r);
	if(l > mid) return query(k * 2 + 1, l, r);
	return query(k * 2, l, mid) + query(k * 2 + 1, mid + 1, r);	
}

int main()
{
	cin >> n >> m >> p;
	for(int i = 1; i <= n; i++){
		cin >> f[i];
		f[i] = f[i] % p;
	}
	build(1, 1, n);
	for(int i = 1; i <= m; i++){
		int a, b, c, d;
		cin >> a;
		if(a == 1){
			cin >> b >> c >> d;
			changeSegment1(1, b, c, d);
		}
		else if(a == 2){
			cin >> b >> c >> d;
			changeSegment2(1, b, c, d);
		}
		else if(a == 3){
			cin >> b >> c;
			ll ans = query(1, b, c) % p;
			cout << ans << endl;
		}
	}
	return 0;
}