线段树30pts求助
查看原帖
线段树30pts求助
91433
czy0407楼主2024/3/4 11:34
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll N=1e6+10;
long long n,m,mod,a[N],tr[N<<2],mark[N<<2],mark2[N<<2],k,x,y,z;
void build(ll l,ll r,ll p)
{
	mark2[p]=1;
	mark[p]=0;
	if(l==r)
	{
		tr[p]=a[l]%mod; 
		return;
	} 
	ll mid=(l+r)/2;
	build(l,mid,p<<1);
	build(mid+1,r,p<<1|1);
	tr[p]=tr[p<<1]+tr[p<<1|1];
	tr[p]%=mod;
}
inline void down(ll p,ll len)
{
	mark[p*2]+=mark[p];
    mark[p*2+1]+=mark[p];
	mark2[p*2]*=mark2[p];
    mark2[p*2+1]*=mark2[p];
	mark[p*2]%=mod;
	mark[p*2+1]%=mod;
    mark2[p*2]%=mod;
    mark2[p*2+1]%=mod;
    tr[p*2]=tr[p*2]*mark2[p]%mod+mark[p]*(len-len/2)%mod;
    tr[p*2+1]=tr[p*2+1]*mark2[p]%mod+mark[p]*(len/2)%mod;
    tr[p*2]%=mod;
    tr[p*2+1]%=mod;
    mark[p]=0;mark2[p]=1;
}
void update(ll l,ll r,ll p)
{
	if(l>y||r<x)return;
	else if(x<=l&&r<=y)
	{
		if(k==2)tr[p]+=(r-l+1)*z;
		else tr[p]*=z;
		tr[p]%=mod;
		if(r>l)
		{
			if(k==2)mark[p]+=z;
			else mark2[p]*=z,mark[p]*=z;
			mark[p]%=mod;
			mark2[p]%=mod;
		}
	}
	else	
	{
		ll mid=(l+r)>>1;
		down(p,r-l+1);
		update(l,mid,p<<1);
		update(mid+1,r,p<<1|1);
		tr[p]=tr[p<<1]+tr[p<<1|1];
		tr[p]%=mod;
	}
}
ll query(ll l,ll r,ll p)
{
	if(l>y||r<x)return 0;
	down(p,r-l+1);
	if(x<=l&&r<=y)return tr[p]%mod;
	ll mid=(l+r)/2;
	return (query(l,mid,p<<1)+query(mid+1,r,p<<1|1))%mod;
}
int main()
{
	scanf("%lld%lld%lld",&n,&m,&mod);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
	build(1,n,1);
	for(int i=1;i<=m;i++)
	{
		scanf("%lld",&k);
		if(k!=3)
		{
			scanf("%lld%lld%lld",&x,&y,&z);
			update(1,n,1);
		}
		else
		{
			scanf("%lld%lld",&x,&y);
			printf("%lld\n",query(1,n,1)%mod); 
		}
	}
	return 0;
}
2024/3/4 11:34
加载中...