线段树

P3373 【模板】线段树 2

题目描述

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

将某区间每一个数乘上 xx

将某区间每一个数加上 xx

求出某区间每一个数的和

输入格式

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

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

接下来 mm 行每行包含若干个整数,表示一个操作,具体如下:

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

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

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

输出格式

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

输入输出样例

输入 #1
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
输出 #1
17
2
做过线段树模板1的都已经了解了线段树的构造,这里就不在赘述。对于懒标记也有点理解了。线段树无非就是考查懒标记的使用,题目要求区间加乘,只要两个懒标记即可,那么如何更新num的值呢?先乘后加还是先加后乘?

  1. 先乘后加:t[n2].num=t[n2].numt[n].cb+t[n].bj(t[n].r-t[n].l+1);
  2. 先加后乘:t[n2].num=t[n].cb(t[n].bj*(t[n].r-t[n].l+1)+t[n].num);

可以看出先加后乘跟正确值不同,是错误的,
所以先乘后加。
最后贴代码,有注释。

#include <iostream>
#define N 100020
#define ll long long
using namespace std;
struct tr {
	int l, r;
	ll num, bj, cb = 1;//bj加法懒标记,cb乘法懒标记初值为1
}t[N*4];//要开四倍空间
ll a[N],p;
void js(ll l, ll r, ll n)//建树
{
	t[n].l = l;
	t[n].r = r;
	if (l == r)
	{
		t[n].num = a[l]%p;
		return;
	}
	ll mid = (l + r) >> 1;
	js(l, mid, n * 2);
	js(mid + 1, r, n * 2 + 1);
	t[n].num = (t[n * 2].num + t[n * 2 + 1].num)%p;
}
void f(ll n)
{
	//if (t[n].bj||t[n].cb>1)不能写,t[n].cb可能为0。本人亲测⌯'ㅅ'⌯
	{
		t[n * 2].num = (t[n].cb*t[n * 2].num+t[n].bj*(t[n * 2].r - t[n * 2].l + 1))%p;//先加后乘。
		t[n * 2+1].num = (t[n].cb*t[n * 2+1].num + t[n].bj*(t[n * 2+1].r - t[n * 2+1].l + 1)) % p;
		t[n * 2].cb = (t[n * 2].cb *t[n].cb) % p;
		t[n * 2 + 1].cb = (t[n * 2+1].cb *t[n].cb) % p;
		t[n * 2].bj =(t[n*2].bj*t[n].cb+ t[n].bj)%p;//要成父节点的乘标记
		t[n * 2 + 1].bj = (t[n * 2+1].bj*t[n].cb + t[n].bj) % p;
		t[n].bj = 0;
		t[n].cb = 1;
	}
}
void qjxc(ll n,ll x, ll y, ll k)//相乘
{
	if (t[n].l >= x && t[n].r <= y)
	{
		
		t[n].cb = (k*t[n].cb)%p;
		t[n].bj = (k*t[n].bj) % p;
		t[n].num=(t[n].num*k)%p;
		return;
	}
	f(n);
	ll mid = (t[n].l + t[n].r) >> 1;
	if (x <= mid) qjxc(n * 2, x, y, k);
	if (y > mid)qjxc(n * 2 + 1, x, y, k);
	t[n].num = (t[n * 2].num + t[n * 2 + 1].num)%p;
}

void xj(ll n, ll x, ll y, ll k)//加
{
	if (t[n].l >= x && t[n].r <= y)
	{
		t[n].bj = (k+t[n].bj) % p;
		t[n].num= (t[n].num+ (t[n].r - t[n].l + 1)*k) % p;
		return;
	}
	f(n);
	ll mid = (t[n].l + t[n].r)>>1;
	if (x <= mid) xj(n * 2, x, y, k);
	if (y > mid)xj(n * 2 + 1, x, y, k);
	t[n].num = (t[n * 2].num + t[n * 2 + 1].num) % p;
}
void sc(int n)//用来测试的,可不写。。。。
{
	f(n);
	if (t[n].l == t[n].r)
	{
		cout << t[n].num<<" ";
		return;
	}
	sc(n * 2);
	sc(n * 2 + 1);
}
ll qjcx(ll n,ll x,ll y)
{
	if (t[n].l >= x && t[n].r <= y)
	{
		return t[n].num%p;
	}
	f(n);
	ll ans = 0;
	ll mid;
	mid = (t[n].r + t[n].l) >>1;
	if (x <= mid) ans =(ans+ qjcx(n * 2, x, y))%p;
	if (y > mid)ans =(ans+ qjcx(n * 2 + 1, x, y))%p;
	return ans%p;
}
int main()
{
	ll n, m;
	scanf("%lld %lld %lld", &n, &m, &p);
	for (int i = 1; i <= n; ++i)scanf("%lld", &a[i]);
	js(1, n, 1);
	for (int i = 1; i <= m; ++i)
	{
		ll x, y, w;
		ll z;
		scanf("%lld", &w);
		if (w > 2) {
			scanf("%lld %lld", &x, &y);
			cout << qjcx(1, x, y) << endl;
			//sc(1), cout << endl;
		}
		else {
			scanf("%lld %lld %lld", &x, &y, &z);
			if (w == 1) qjxc(1, x, y, z);// , sc(1), cout << endl;
			else xj(1, x, y, z);//, sc(1) , cout << endl;
		}
	}
	return 0;
}
快可灵

。。。。。。。。。٩(* ఠO ఠ)=3⁼³₌₃⁼³₌₃⁼³₌₃嘟啦啦啦啦

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值