线段树板子样例不过玄关求调
查看原帖
线段树板子样例不过玄关求调
1364148
yangzimu0131楼主2024/8/5 16:53
#include<bits/stdc++.h>
using namespace std;
int mod,a[100010];
struct node{
	int sum,mul,add;
}tree[100010<<2];
void build(int k,int l,int r)
{
	tree[k].mul=1;
	if(l==r)
	{
		tree[k].sum=a[l]%mod;
		return ;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	tree[k].sum=(tree[k<<1].sum+tree[k<<1|1].sum)%mod;
}
void down(int k,int l,int r,int mul,int add)
{
	tree[k].sum=(tree[k].sum*mul+add*(r-l+1))%mod;
	tree[k].mul=(tree[k].mul*mul)%mod;
	tree[k].add=(tree[k].add*mul+add)%mod;
}
void push(int k,int l,int r)
{
	int mid=(l+r)>>1;
	down(k<<1,l,mid,tree[k].mul,tree[k].add);
	down(k<<1|1,mid+1,r,tree[k].mul,tree[k].add);
	tree[k].mul=1;
	tree[k].add=0;
}
void mul(int k,int l,int r,int x,int y,int temp)
{
	if(x<=l&&r<=y)
	{
		down(k,l,r,temp,0);
		return ;
	}
	push(k,l,r);
	int mid=(l+r)>>1;
	if(x<=mid)
		mul(k<<1,l,mid,x,y,temp);
	if(mid<y)
		mul(k<<1|1,mid+1,r,x,y,temp);
	tree[k].sum=(tree[k<<1].sum+tree[k<<1|1].sum)%mod;
}
void add(int k,int l,int r,int x,int y,int temp)
{
	if(x<=l&&r<=y)
	{
		down(k,l,r,1,temp);
		return ;
	}
	push(k,l,r);
	int mid=(l+r)>>1;
	if(x<=mid)
		add(k<<1,l,mid,x,y,temp);
	if(mid<y)
		add(k<<1|1,mid+1,r,x,y,temp);
	tree[k].sum=(tree[k<<1].sum+tree[k<<1|1].sum)%mod;
}
int query(int k,int l,int r,int x,int y)
{
	if(x<=l&&r<=y)
		return tree[k].sum;
	push(k,l,r);
	int mid=(l+r)>>1,ans=0;
	if(x<=mid)
		ans+=query(k<<1,l,mid,x,y);
	if(mid<y)
		ans+=query(k<<1|1,mid+1,r,x,y);
}
signed main()
{
	int n,m,opt,x,y,k;
	cin>>n>>m>>mod;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=m;i++)
	{
		cin>>opt;
		if(opt==1)
		{
			cin>>x>>y>>k;
			mul(1,1,n,x,y,k);
		}else if(opt==2)
		{
			cin>>x>>y>>k;
			add(1,1,n,x,y,k);
		}else{
			cin>>x>>y;
			cout<<query(1,1,n,x,y)<<endl;
		}
	}
	return 0;
}
2024/8/5 16:53
加载中...