线段树麻烦大佬差错,RE*10
查看原帖
线段树麻烦大佬差错,RE*10
51135
星之海楼主2019/2/18 00:09
#define maxn 100010
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int a[maxn+2];
int mod;
struct tree
{
	int l,r;
	long long pre,add;
}t[4*maxn+2];
void build(int p,int l,int r)
{
	t[p].l=l;
	t[p].r=r;
	if(l==r)
	{
		t[p].pre=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	t[p].pre=t[p*2].pre+t[p*2+1].pre;
}
void spread(int p)
{
	if(t[p].add)
	{
		t[p*2].pre+=t[p].add*(t[p*2].r-t[p*2].l+1);
		t[p*2+1].pre+=t[p].add*(t[p*2+1].r-t[p*2+1].l+1);
		t[p*2].add+=t[p].add;
		t[p*2+1].add+=t[p].add;
		t[p].add=0;
	}
}
void changejia(int p,int x,int y,int z)
{
	if(x<=t[p].l&&y>=t[p].r)
	{
		t[p].pre+=(long long)z*(t[p].r-t[p].l+1);
		t[p].add+=z;
		return;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(x<=mid)
	changejia(p*2,x,y,z);
	if(y>mid)
	changejia(p*2+1,x,y,z);
	t[p].pre=t[p*2].pre+t[p*2+1].pre;
}
void changecheng(int p,int x,int y,int z)
{
	if(x<=t[p].l&&y>=t[p].r)
	{
		t[p].pre+=(long long)z*(t[p].r-t[p].l+1);
		t[p].add*=z;
		return;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(x<=mid)
	changecheng(p*2,x,y,z);
	if(y>mid)
	changecheng(p*2,x,y,z);
	t[p].pre=t[p*2].pre+t[p*2+1].pre;
}
long long ask(int p,int x,int y)
{
	if(x<=t[p].l&&y>=t[p].r)
	return t[p].pre;
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	long long ans=0;
	if(x<=mid)
	ans=(ans+ask(p*2,x,y))%mod;
	if(y>mid)
	ans=(ans+ask(p*2+1,x,y))%mod;
	return ans%mod;
}
int main()
{
	int i,n,m;
	cin>>n>>m>>mod;
	for(i=1;i<=n;++i)
	cin>>a[i];
	build(1,1,n);
	for(i=1;i<=m;++i)
	{
		int q,x,y,z;
		cin>>q;
		if(q==1)
		{
			cin>>x>>y>>z;
			changecheng(1,x,y,z);
		}
		else
		if(q==2)
		{
			cin>>x>>y>>z;
			changejia(1,x,y,z);
		}
		else
		{
			cin>>x>>y;
			cout<<ask(1,x,y)%mod<<endl;
		}
	}
	return 0;
}
2019/2/18 00:09
加载中...