P3373 【模板】线段树 2

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,q,a[1000005],m;
struct node {
	int sum,l,r,mu,add;
} t[1000005];
void build(int i,int l,int r) {
	t[i].l=l,t[i].r=r;
	t[i].mu=1;
	if(l==r) {
		t[i].sum=a[l]%m;
		return ;
	}
	int mid=(l+r)/2;
	build(i*2,l,mid);
	build(i*2+1,mid+1,r);
	t[i].sum=(t[i*2].sum+t[i*2+1].sum)%m;
}
void spread(int i) {
	t[i*2].sum=(t[i].mu*t[i*2].sum+((t[i*2].r-t[i*2].l+1)*t[i].add)%m)%m;
	t[i*2+1].sum=(t[i].mu*t[i*2+1].sum+(t[i].add*(t[i*2+1].r-t[i*2+1].l+1))%m)%m;

	t[i*2].mu=(t[i*2].mu*t[i].mu)%m;
	t[i*2+1].mu=(t[i*2+1].mu*t[i].mu)%m;

	t[i*2].add=(t[i*2].add*t[i].mu+t[i].add)%m;
	t[i*2+1].add=(t[i*2+1].add*t[i].mu+t[i].add)%m;

	t[i].mu=1,t[i].add=0;
}
void add(int i,int l,int r,int k) {
	if(t[i].l>=l&&t[i].r<=r) {
		t[i].add=(t[i].add+k)%m;
		t[i].sum=(t[i].sum+k*(t[i].r-t[i].l+1))%m;
		return ;
	}
	spread(i);
	t[i].sum=(t[i*2].sum+t[i*2+1].sum)%m;
	int mid=(t[i].l+t[i].r)/2;
	if(l<=mid) add(i*2,l,r,k);
	if(mid<r) add(i*2+1,l,r,k);
	t[i].sum=(t[i*2].sum+t[i*2+1].sum)%m;

}
void mu(int i,int l,int r,int k) {
	if(t[i].l>=l&&t[i].r<=r) {
		t[i].add=(t[i].add*k)%m;
		t[i].mu=(t[i].mu*k)%m;
		t[i].sum=(t[i].sum*k)%m;
		return;
	}
	spread(i);
	t[i].sum=t[i*2].sum+t[i*2+1].sum;
	int mid=(t[i].l+t[i].r)/2;
	if(t[i*2].r>=l) mu(i*2,l,r,k);
	if(t[i*2+1].l<=r) mu(i*2+1,l,r,k);
	t[i].sum=(t[i*2].sum+t[i*2+1].sum)%m;
}
int query(int i,int l,int r) {
	if(t[i].l>=l&&t[i].r<=r) return t[i].sum;
	spread(i);
	int ans=0;
	int mid=(t[i].l+t[i].r)/2;
	if(t[i*2].r>=l) ans=(ans+query(i*2,l,r))%m;
	if(t[i*2+1].l<=r) ans=(ans+query(i*2+1,l,r))%m;
	return ans;
}
main() {
	cin>>n>>q>>m;
	for(int i=1; i<=n; i++) cin>>a[i];
	build(1,1,n);
	while(q--) {
		int op;
		cin>>op;
		if(op==1) {
			int l,r,k;
			cin>>l>>r>>k;
			mu(1,l,r,k);
		} else if(op==2) {
			int l,r,k;
			cin>>l>>r>>k;
			add(1,l,r,k);
		} else {
			int l,r;
			cin>>l>>r;
			cout<<query(1,l,r)<<'\n';
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值