P3373(线段树+(乘法,加法混合操作))

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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值