BZOJ 1798 线段树 双LAZY操作

线段树裸题  

双LAZY操作  记录 mul和add

维护 a*x+b即可

#include "stdio.h"
#include "string.h"
#include "math.h"

struct comp
{
	int l,r;
	__int64   sum,add,mul;
	int size()
	{
		return r-l+1;
	}
} data[500010];

__int64  mod;

void build(int l,int r,int k)
{
	int mid;

	data[k].l=l;
	data[k].r=r;
	data[k].add=0;
	data[k].mul=1;

	if (l==r)
	{
		scanf("%I64d",&data[k].sum);
		data[k].sum%=mod;
		return ;
	}

	mid=(l+r)/2;

	build(l,mid,k*2);
	build(mid+1,r,k*2+1);

	data[k].sum=(data[k*2].sum+data[k*2+1].sum)%mod;

	return ;
}

void Pushdown(int k)
{
	data[k*2].mul=(data[k*2].mul * data[k].mul) % mod;
	data[k*2].add=((data[k*2].add * data[k].mul) % mod + data[k].add  ) % mod;
	data[k*2].sum=(data[k*2].sum * data[k].mul % mod + data[k].add * data[k*2].size() % mod) % mod;

	data[k*2+1].mul=(data[k*2+1].mul * data[k].mul) % mod;
	data[k*2+1].add=((data[k*2+1].add * data[k].mul) % mod + data[k].add ) % mod;
	data[k*2+1].sum=(data[k*2+1].sum * data[k].mul % mod + data[k].add * data[k*2+1].size() % mod) % mod;

	data[k].mul=1;
	data[k].add=0;
}



__int64 search(int l,int r,int k)
{
	int mid;

	if (l==data[k].l && r==data[k].r)
		return data[k].sum;

	Pushdown(k);

	mid=(data[k].l+data[k].r)/2;

	if (r<=mid) return search(l,r,k*2);
	else 
		if (l>mid) return search(l,r,k*2+1);
		else 
			return search(l,mid,k*2)+search(mid+1,r,k*2+1);
}

void updata(int l,int r,int k,int key,int op)
{
	int mid;
	if (l==data[k].l && data[k].r==r)
	{
		if (op==1)
		{
			data[k].sum=data[k].sum * key % mod;
			data[k].mul=(data[k].mul * key) % mod;
			data[k].add=(data[k].add * key) % mod;
		}
		else 
		{
			data[k].sum=(data[k].sum + key * data[k].size() % mod) % mod;
			data[k].add=(data[k].add + key) % mod;
		}
		return ;
	}

	Pushdown(k);

	mid=(data[k].l+data[k].r)/2;

	if (r<=mid) updata(l,r,k*2,key,op);
	else 
		if (l>mid) updata(l,r,k*2+1,key,op);
		else 
		{
			updata(l,mid,k*2,key,op);
			updata(mid+1,r,k*2+1,key,op);
		}

	data[k].sum=(data[k*2].sum+data[k*2+1].sum) % mod;

	return ;
}

int main()
{
	int n,m,op,a,b;
	__int64   c;
	scanf("%d%d",&n,&mod);
	build(1,n,1);

	scanf("%d",&m);
	while (m--)
	{
		scanf("%d",&op);
		if (op==3)
		{
			scanf("%d%d",&a,&b);
			printf("%I64d\n",search(a,b,1)%mod);
		}
		else 
		{
			scanf("%d%d%lld",&a,&b,&c);
			c%=mod;
			updata(a,b,1,c,op);
		}
	}
	return 0;
}