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的值呢?先乘后加还是先加后乘?
- 先乘后加:t[n2].num=t[n2].numt[n].cb+t[n].bj(t[n].r-t[n].l+1);
- 先加后乘: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⁼³₌₃⁼³₌₃⁼³₌₃嘟啦啦啦啦