线段树2

来源:https://www.luogu.org/problemnew/show/P3373#sub

 

看题!!!!!!

别把n m看反了!!!!!!

re错误:数组下标越界。。。爆内存。。。

以后看到样例是n=m的要特别注意!!!

#include <cstdio>
#include <algorithm>
#include <fstream>
#include <cstring>

using namespace std;

struct node
{
	int l,r;
	long long add,sum,mul;
};

#define lson 2*u
#define rson 2*u+1
const int maxSize=100000;
long long m,n,p,sum,a[maxSize+5];
node N[4*maxSize+5];

void pushup(int u)
{
	N[u].sum=(N[lson].sum+N[rson].sum)%p;
	return ;
}

void update(int u)
{
	if (N[u].mul!=1)
	{
		N[lson].sum=(N[lson].sum*N[u].mul)%p;
		N[rson].sum=(N[rson].sum*N[u].mul)%p;
		
		N[lson].mul=(N[lson].mul*N[u].mul)%p;
		N[rson].mul=(N[rson].mul*N[u].mul)%p;
		
		N[lson].add=(N[lson].add*N[u].mul)%p;
		N[rson].add=(N[rson].add*N[u].mul)%p;
		N[u].mul=1;
	}
	
	if (N[u].add!=0)
	{
		N[lson].add=(N[lson].add+N[u].add)%p;
		N[rson].add=(N[rson].add+N[u].add)%p;
		
		N[lson].sum=(N[lson].sum+N[u].add*(N[lson].r-N[lson].l+1))%p;
		N[rson].sum=(N[rson].sum+N[u].add*(N[rson].r-N[rson].l+1))%p;
		N[u].add=0;
	}
	return ;
}

void build(int u,int begin,int end)
{
	int mid;
	
	N[u].l=begin;	N[u].r=end;		N[u].mul=1;		N[u].add=0;
	if (begin==end)
	{
		N[u].sum=a[begin];
		return ;
	}
	
	mid=(begin+end)/2;
	
	build(lson,begin,mid);
	build(rson,mid+1,end);
	pushup(u);
}

void mul(int u,int begin,int end,long long k)
{
	int mid;
	
	if (N[u].l==begin && N[u].r==end)
	{
		N[u].sum=(N[u].sum*k)%p;
		N[u].mul=(N[u].mul*k)%p;
		N[u].add=(N[u].add*k)%p;
		return ;
	}
	
	update(u);
	
	mid=(N[u].r+N[u].l)/2;
	if (end<=mid)
		mul(lson,begin,end,k);
	else if (begin>mid)
		mul(rson,begin,end,k);
	else
	{
		mul(lson,begin,mid,k);
		mul(rson,mid+1,end,k);
	}
	pushup(u);
}

void add(int u,int begin,int end,long long k)
{
	int mid;
	
	if (N[u].l==begin && N[u].r==end)
	{
		N[u].sum=(N[u].sum+k*(end-begin+1))%p;
		N[u].add=(N[u].add+k)%p;
		return ;
	}
	
	update(u);
	
	mid=(N[u].r+N[u].l)/2;
	if (end<=mid)
		add(lson,begin,end,k);
	else if (begin>mid)
		add(rson,begin,end,k);
	else
	{
		add(lson,begin,mid,k);
		add(rson,mid+1,end,k);
	}
	pushup(u);	
}

void query(int u,int begin,int end)
{
	int mid;
	
	if (N[u].l==begin && N[u].r==end)
	{
		sum=(sum+N[u].sum)%p;
		return ;
	}
	
	update(u);
	mid=(N[u].l+N[u].r)/2;
	if (end<=mid)
		query(lson,begin,end);
	else if (begin>mid)
		query(rson,begin,end);
	else
	{
		query(lson,begin,mid);
		query(rson,mid+1,end);
	}
}

int main()
{
	long long i,k,x,y,z;
	
	freopen("a.txt","r",stdin);
	scanf("%lld%lld%lld",&n,&m,&p);
	
	for (i=1;i<=m;i++)
	{
		scanf("%lld",&a[i]); 
		a[i]%=p;
	}
	
	build(1,1,m);
	for (i=0;i<n;i++)
	{
		scanf("%lld",&k);
		if (k==1)
		{
			scanf("%lld%lld%lld",&x,&y,&z);
			z=z%p;
			mul(1,x,y,z);
		}
		else if (k==2)
		{
			scanf("%lld%lld%lld",&x,&y,&z);
			z=z%p;
			add(1,x,y,z);
		}
		else
		{
			sum=0;
			scanf("%lld%lld",&x,&y);
			query(1,x,y);
			printf("%lld\n",sum);
		}
	}
	
	
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值