萌新刚学线段树,样例没过求调
查看原帖
萌新刚学线段树,样例没过求调
229957
Wu_while楼主2021/7/18 15:57

RT

#include<iostream>
#include<cstdio>
#define MAXN 100010
using namespace std;
int n,m,MOD;
int op,x,y,v;
int a[MAXN];
struct node
{
	int sum;
	int plue;
	int mul;
}
t[MAXN*4];
void push_up(int k)
{
	t[k].sum=t[k<<1].sum+t[k<<1|1].sum;
	t[k].sum%=MOD;
}
void add(int k,int l,int r,int a,int b)
{
	t[k].sum*=a;
	t[k].sum%=MOD;
	t[k].sum+=b*(r-l+1);
	t[k].sum%=MOD;
	t[k].mul*=a;
	t[k].mul%=MOD;
	t[k].plue*=a;
	t[k].plue%=MOD;
	t[k].plue+=b;
	t[k].plue%=MOD;
}
void push_down(int k,int l,int r)
{
	int k1=k<<1,k2=k<<1|1;
	int mid=(l+r)>>1;
	add(k<<1,l,mid,t[k].mul,t[k].plue);
	add(k<<1|1,mid+1,r,t[k].mul,t[k].plue);
	t[k].plue=0;
	t[k].mul=1;
}
void build(int k,int l,int r)
{
	t[k].mul=1;
	if(l==r)
	{
		t[k].sum=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	push_up(k);
}
void update_plue(int k,int l,int r,int L,int R,int v)
{
	if(R<l||L>r)
		return;
	if(L<=l&&R>=r)
	{
		t[k].plue+=v;
		t[k].plue%=MOD;
		t[k].sum+=v*(r-l+1)%MOD;
		t[k].sum%=MOD;
		return;
	}
	push_down(k,l,r);
	int mid=(l+r)>>1;
	update_plue(k<<1,l,mid,L,R,v);
	update_plue(k<<1|1,mid+1,r,L,R,v);
	push_up(k);
}
void update_mul(int k,int l,int r,int L,int R,int v)
{
	if(R<l||L>r)
		return;
	if(L<=l&&R>=r)
	{
		t[k].mul*=v;
		t[k].mul%=MOD;
		t[k].plue*=v;
		t[k].plue%=MOD;
		t[k].sum*=v;
		t[k].sum%=MOD;
		return;
	}
	push_down(k,l,r);
	int mid=(l+r)>>1;
	update_plue(k<<1,l,mid,L,R,v);
	update_plue(k<<1|1,mid+1,r,L,R,v);
	push_up(k);
}
int query(int k,int l,int r,int L,int R)
{
	if(R<l||L>r)
		return 0;
	if(L<=l&&R>=r)
		return t[k].sum;
	push_down(k,l,r);
	int mid=(l+r)>>1;
	int s1,s2;
	s1=query(k<<1,l,mid,L,R);
	s2=query(k<<1|1,mid+1,r,L,R);
	return (s1+s2)%MOD; 
}
int main()
{
	scanf("%d%d%d",&n,&m,&MOD);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d%d",&x,&y,&v);
			update_mul(1,1,n,x,y,v);
		}
		else if(op==2)
		{
			scanf("%d%d%d",&x,&y,&v);
			update_plue(1,1,n,x,y,v);
		}
		else
		{
			scanf("%d%d",&x,&y);
			printf("%d\n",query(1,1,n,x,y));
		}
	}
	return 0;
}
2021/7/18 15:57
加载中...