【模板】线段树 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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值