线段树--双标签

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn  100010
typedef long long LL;
using namespace std;
LL a[maxn],n,p;
struct node{
    int l,r;
	LL sum,alltag,multag;
}tr[maxn*4];
void build(int id,int l,int r);
void change(int id,int ql,int qr,LL a,LL m);
LL query(int id,int ql,int qr);
void pushdown(int id);
int main(){
	//freopen("in.txt","r",stdin);
    int i,m;
	scanf("%lld %lld",&n,&p);
	for(i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	cin>>m;
	build(1,1,n);
	while(m--){
		LL o,a,b,c;
		scanf("%lld",&o);
		if(o==1){
			scanf("%lld %lld %lld",&a,&b,&c);
			change(1,a,b,0,c);
		}
		if(o==2){
			scanf("%lld %lld %lld",&a,&b,&c);
			change(1,a,b,c,1);
		}
		if(o==3){
			scanf("%lld %lld",&a,&b);
			printf("%lld\n",query(1,a,b));
		}
	}
}
void build(int id,int l,int r){
	tr[id].l=l; tr[id].r=r;
    tr[id].alltag=0;
	tr[id].multag=1;
	if(l==r){
		tr[id].sum=a[l]%p;
		return;
	}
	int mid=(l+r)/2;
	build(id<<1,l,mid);
	build(id<<1|1,mid+1,r);
	tr[id].sum=(tr[id<<1].sum+tr[id<<1|1].sum)%p;
}
void change(int id,int ql,int qr,LL a,LL m){
	int l=tr[id].l,r=tr[id].r;
	if(ql<=l&&qr>=r){
		tr[id].sum=(tr[id].sum*m%p+a*(r-l+1)%p)%p;
		tr[id].multag=tr[id].multag*m%p;
		tr[id].alltag=(tr[id].alltag*m%p+a)%p;
		return;
	}
	pushdown(id);
	int mid=(l+r)/2;
    if(ql<=mid)
	    change(id<<1,ql,qr,a,m);
	if(qr>mid)
	    change(id<<1|1,ql,qr,a,m);
	tr[id].sum=(tr[id<<1].sum+tr[id<<1|1].sum)%p;
}
void pushdown(int id){
	LL m=tr[id].multag,a=tr[id].alltag;
	int l=tr[id].l,r=tr[id].r;
	int mid=(l+r)/2;
	tr[id<<1].sum=(tr[id<<1].sum*m%p+a*(mid-l+1)%p)%p;
	tr[id<<1|1].sum=(tr[id<<1|1].sum*m%p+a*(r-mid)%p)%p;
	tr[id<<1].multag=tr[id<<1].multag*m%p;
	tr[id<<1|1].multag=tr[id<<1|1].multag*m%p;
	tr[id<<1].alltag=(tr[id<<1].alltag*m%p+a)%p;
	tr[id<<1|1].alltag=(tr[id<<1|1].alltag*m%p+a)%p;
	tr[id].multag=1;
	tr[id].alltag=0;
}
LL query(int id,int ql,int qr){
	int l=tr[id].l,r=tr[id].r;
	if(ql<=l&&qr>=r){
		return tr[id].sum;
	}
	pushdown(id);
	LL p1=0,p2=0;
	int mid=(l+r)/2;
	if(ql<=mid)
	    p1=query(id<<1,ql,qr);
	if(qr>mid)
	    p2=query(id<<1|1,ql,qr);
	tr[id].sum=(tr[id<<1].sum+tr[id<<1|1].sum)%p;
	return (p1+p2)%p;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值