线段树模版

维护序列-区间乘与区间加

题目描述

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。

有长为N的数列，不妨设为a1,a2,…,aN 。

有如下三种操作形式：

(1)把数列中的一段数全部乘一个值;

(2)把数列中的一段数全部加一个值;

(3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。

输入格式

第一行两个整数N和P(1≤P≤1000000000）。

第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。

第三行有一个整数M，表示操作总数。从第四行开始每行描述一个操作，输入的操作有以下三种形式：

操作1：“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c (1≤t≤g≤N,0≤c≤1000000000)。

操作2：“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。

操作3：“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

线段树开long long容易超时！！！

#include<bits/stdc++.h>
//#define int long long
#define ll long long
#define ull /*unsigned*/ long long
#define fd(i,a,b) for(int i=a,_i=b;i<=_i;i=-~i)
#define bd(i,a,b) for(int i=a,_i=b;i>=_i;i=~-i)
#define Db(a,b,ans) cout<<a<<';'<<b<<','<<ans<<endl
#define debug puts("-----------");
using namespace std;

const int N=1e6+509;
ll n,m,a[N],mod;

struct ST
{
	
	struct node
	{
		ll l,r,add,sum,mul;
		#define l(x) st[x].l
		#define r(x) st[x].r
		#define add(x) st[x].add
		#define sum(x) st[x].sum
		#define mul(x) st[x].mul
		#define ls(x) (x<<1)
		#define rs(x) (x<<1|1)
	}st[N<<1];
	
	inline void pushdown(int p)
	{
		if(add(p)==0&&mul(p)==1) return;
		sum(ls(p))=(sum(ls(p))*mul(p)%mod+add(p)*(r(ls(p))-l(ls(p))+1)%mod)%mod;
		sum(rs(p))=(sum(rs(p))*mul(p)%mod+add(p)*(r(rs(p))-l(rs(p))+1)%mod)%mod;
		mul(ls(p))=mul(ls(p))*mul(p)%mod;
		mul(rs(p))=mul(rs(p))*mul(p)%mod;
		add(ls(p))=(add(ls(p))*mul(p)%mod+add(p))%mod;
		add(rs(p))=(add(rs(p))*mul(p)%mod+add(p))%mod;
		add(p)=0;
		mul(p)=1;
	}
	
	inline void pushup(int p)
	{
		sum(p)=sum(ls(p))+sum(rs(p));
		sum(p)%=mod;
	}
	
	void build(int p,int l,int r)
	{
		l(p)=l,r(p)=r;
		add(p)=0,mul(p)=1;
		if(l==r)
		{
			sum(p)=a[r];
			return;
		}
		int mid=((r-l)>>1)+l;
		build(ls(p),l,mid);
		build(rs(p),mid+1,r);
		pushup(p);
	}
	
	void Add(int p,int l,int r,ll v)
	{
		if(l(p)>=l&&r(p)<=r)
		{
			add(p)=(add(p)+v)%mod;
			sum(p)=(sum(p)+v*(r(p)-l(p)+1)%mod)%mod;
			return;
		}
		pushdown(p);
		int mid=((r(p)-l(p))>>1)+l(p);
		if(l<=mid)	Add(ls(p),l,r,v);
		if(mid<r)	Add(rs(p),l,r,v);
		pushup(p);
	}
	
	void Mul(int p,int l,int r,ll v)
	{
		if(l(p)>=l&&r(p)<=r)
		{
			add(p)=(add(p)*v)%mod;
			mul(p)=(mul(p)*v)%mod;
			sum(p)=(sum(p)*v)%mod;
			return;
		}
		pushdown(p);
		int mid=((r(p)-l(p))>>1)+l(p);
		if(l<=mid)	Mul(ls(p),l,r,v);
		if(mid<r)	Mul(rs(p),l,r,v);
		pushup(p);
	}
	
	ll ask(int p,int l,int r)
	{
		if(l(p)>=l&&r(p)<=r) return sum(p);
		pushdown(p);
		int mid=((r(p)-l(p))>>1)+l(p);
		ll res=0;
		if(l<=mid)	res+=ask(ls(p),l,r)%mod;
		if(mid<r)	res+=ask(rs(p),l,r)%mod;
		return res%mod;
	}
	
}St;

signed main()
{
#ifdef FJ
	freopen("seq.in","r",stdin);
	freopen("seq.out","w",stdout);
#endif
//	ios::sync_with_stdio(0);
//	cin.tie(0); cout.tie(0);
	
	scanf("%lld%lld",&n,&mod);
	
	fd(i,1,n) scanf("%lld",&a[i]);
		
	St.build(1,1,n);
	
	scanf("%lld",&m);
	ll op,t,g,c;
	
	fd(i,1,m)
	{
		
		scanf("%lld",&op);
		
		if(op==1)
		{
			scanf("%lld%lld%lld",&t,&g,&c);
			St.Mul(1,t,g,c);
		}
		
		if(op==2)
		{
			scanf("%lld%lld%lld",&t,&g,&c);
			St.Add(1,t,g,c);
		}
		
		if(op==3)
		{
			scanf("%lld%lld",&t,&g);
			printf("%lld\n",St.ask(1,t,g));
		}
		
	}
	
	return 0;
}