P3373 模板线段树 2区间乘/加 区间查询

Posted 2021-11-23 辉小歌

https://www.luogu.com.cn/problem/P3373

#include<bits/stdc++.h>
using namespace std;
typedef long long int LL;
const int N=1e5+10; 
int n,m,p;
int w[N];
struct node
{
	int l,r;
	int sum,add,mul;
}tr[N*4];
void pushup(int u)
{
	tr[u].sum=(tr[u<<1].sum+tr[u<<1 | 1].sum)%p;
}
void eval(node &t,int add,int mul)
{
	t.sum=((LL)t.sum*mul+(LL)(t.r-t.l+1)*add)%p;
	t.mul=(LL)t.mul*mul%p;
	t.add=((LL)t.add*mul+add)%p;
}
void pushdown(int u)
{
	eval(tr[u<<1],tr[u].add,tr[u].mul);
	eval(tr[u<<1 | 1],tr[u].add,tr[u].mul);
	tr[u].add=0,tr[u].mul=1;
}
void build(int u,int l,int r)
{
	if(l==r) tr[u]={l,r,w[r],0,1};
	else
	{
		tr[u]={l,r,0,0,1};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		pushup(u);
	}
}
void modify(int u,int l,int r,int add,int mul)
{
	if(tr[u].l>=l&&tr[u].r<=r) eval(tr[u],add,mul);
	else
	{
		pushdown(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1,l,r,add,mul);
		if(r>mid) modify(u<<1|1,l,r,add,mul);
		pushup(u);
	}
}
int query(int u,int l,int r)
{
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
	pushdown(u);
	int mid=tr[u].l+tr[u].r>>1;
	int sum=0;
	if(l<=mid) sum=query(u<<1,l,r);
	if(r>mid) sum=(sum+query( u<< 1 | 1 ,l,r) )%p;
	return sum;
}
int main(void)
{
	cin>>n>>m>>p;
	for(int i=1;i<=n;i++) cin>>w[i];
	build(1,1,n);
	while(m--)
	{
	    int op; cin>>op;
	    if(op==1)//区间乘法
	    {
	        int l,r,d; cin>>l>>r>>d;
	        modify(1,l,r,0,d);
	    }
	    else if(op==2)//区间加
	    {
	        int l,r,d; cin>>l>>r>>d;
	        modify(1,l,r,d,1);
	    }
	    else//区间查询
	    {
	        int l,r; cin>>l>>r;
	        cout<<query(1,l,r)<<endl;
	    }
	}
	return 0;
}