题目描述
如题,已知一个数列,你需要进行下面三种操作:
- 将某区间每一个数乘上 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;
}