P3373
题目描述
如题,已知一个数列,你需要进行下面三种操作:
将某区间每一个数乘上 x
将某区间每一个数加上 x
求出某区间每一个数的和
输入格式
第一行包含三个整数 n,m,p,分别表示该数列数字的个数、操作的总个数和模数
第二行包含 n个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。
接下来 m 行每行包含若干个整数,表示一个操作,具体如下:
操作 1: 格式:1 x y k 含义:将区间 [x,y] 内每个数乘上 k
操作 2: 格式:2 x y k 含义:将区间 [x,y] 内每个数加上 k
操作 3: 格式:3 x y 含义:输出区间 [x,y] 内每个数的和对 p 取模所得的结果
题解:以乘法为分析的核心,然后乘法mul[]数组的变化会引起add[]数组的变化,故在乘法更新的过程中还要顺带更新add[]数组,详细操作见代码
#include <iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=1e5+10;
int n,m;
LL p;
LL a[maxn],add[maxn<<2],mul[maxn<<2];
LL sum[maxn<<2];
void build(int l,int r,int rt)
{
if(l==r){
sum[rt]=a[l]%p;
add[rt]=0;
mul[rt]=1;
return ;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
add[rt]=0;
mul[rt]=1;
}
void pushdown(int l,int r,int rt)
{
int mid=(l+r)>>1;
sum[rt<<1]=(sum[rt<<1]*mul[rt]+(mid-l+1)*add[rt])%p;
sum[rt<<1|1]=(sum[rt<<1|1]*mul[rt]+(r-mid)*add[rt])%p;
mul[rt<<1]=(mul[rt<<1]*mul[rt])%p;
mul[rt<<1|1]=(mul[rt<<1|1]*mul[rt])%p;
add[rt<<1]=(add[rt<<1]*mul[rt]+add[rt])%p;
add[rt<<1|1]=(add[rt<<1|1]*mul[rt]+add[rt])%p;
add[rt]=0;
mul[rt]=1;
}
LL query(int l,int r,int rt,int x,int y)
{
if(l>=x&&r<=y){
return sum[rt];
}
pushdown(l,r,rt);
int mid=(l+r)>>1;
LL lsum=0,rsum=0;
if(x<=mid){
lsum=query(l,mid,rt<<1,x,y);
}
if(y>mid){
rsum=query(mid+1,r,rt<<1|1,x,y);
}
return (lsum+rsum)%p;
}
void update1(int l,int r,int rt,int x,int y,LL d)
{
if(l>=x&&r<=y){
mul[rt]=(mul[rt]*d)%p;
sum[rt]=(sum[rt]*d)%p;
add[rt]=(add[rt]*d)%p;
return ;
}
pushdown(l,r,rt);
int mid=(l+r)>>1;
if(x<=mid){
update1(l,mid,rt<<1,x,y,d);
}
if(y>mid){
update1(mid+1,r,rt<<1|1,x,y,d);
}
sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
}
void update2(int l,int r,int rt,int x,int y,LL d)
{
if(l>=x&&r<=y){
sum[rt]=(sum[rt]+(r-l+1)*d)%p;
add[rt]=(add[rt]+d)%p;
return ;
}
pushdown(l,r,rt);
int mid=(l+r)>>1;
if(x<=mid){
update2(l,mid,rt<<1,x,y,d);
}
if(y>mid){
update2(mid+1,r,rt<<1|1,x,y,d);
}
sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
}
int main()
{
scanf("%d%d%lld",&n,&m,&p);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
}
build(1,n,1);
int op,x,y;
LL k;
while(m--){
scanf("%d",&op);
if(op==1){
scanf("%d%d%lld",&x,&y,&k);
update1(1,n,1,x,y,k);
}
else
if(op==2){
scanf("%d%d%lld",&x,&y,&k);
update2(1,n,1,x,y,k);
}
else{
scanf("%d%d",&x,&y);
printf("%lld\n",query(1,n,1,x,y));
}
}
return 0;
}