【Luogu AHOI2009】维护序列

[AHOI2009] 维护序列

题目背景

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

题目描述

有一个长为 $n$ 的数列 { a n } \{a_n\} {an​}，有如下三种操作形式：

1. 格式 1 t g c，表示把所有满足 $t\le i\le g$ 的 $a_i$ 改为 $a_i\times c$ ;
2. 格式 2 t g c 表示把所有满足 $t\le i\le g$ 的 $a_i$ 改为 $a_i+c$ ;
3. 格式 3 t g 询问所有满足 $t\le i\le g$ 的 $a_i$ 的和，由于答案可能很大，你只需输出这个数模 $p$ 的值。

输入格式

第一行两个整数 $n$ 和 $p$。

第二行含有 $n$ 个非负整数，表示数列 { a i } \{a_i\} {ai​} 。

第三行有一个整数 $m$，表示操作总数。

从第四行开始每行描述一个操作，同一行相邻两数之间用一个空格隔开，每行开头和末尾没有多余空格。

输出格式

对每个操作 3，按照它在输入中出现的顺序，依次输出一行一个整数表示询问结果。

样例 #1

样例输入 #1

7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7

样例输出 #1

2
35
8

提示

样例输入输出 1 解释

• 初始时数列为 { 1 , 2 , 3 , 4 , 5 , 6 , 7 } \{1,2,3,4,5,6,7\} {1,2,3,4,5,6,7}。
• 经过第 $1$ 次操作后，数列为 { 1 , 10 , 15 , 20 , 25 , 6 , 7 } \{1,10,15,20,25,6,7\} {1,10,15,20,25,6,7}。
• 对第 $2$ 次操作，和为 $10+15+20=45$，模 $43$ 的结果是 $2$。
• 经过第 $3$ 次操作后，数列为 { 1 , 10 , 24 , 29 , 34 , 15 , 16 } \{1,10,24,29,34,15,16\} {1,10,24,29,34,15,16}。
• 对第 $4$ 次操作，和为 $1+10+24=35$，模 $43$ 的结果是 $35$。
• 对第 $5$ 次操作，和为 $29+34+15+16=94$，模 $43$ 的结果是$8$。

数据规模与约定

测试数据规模如下表所示：

数据点编号	1	2	3	4	5	6	7	8	9,10
$n=$	$10$	$1000$	$1000$	$10000$	$60000$	$70000$	$80000$	$90000$	$100000$
$m=$	$10$	$1000$	$1000$	$10000$	$60000$	$70000$	$80000$	$90000$	$100000$

对于全部的测试点，保证 $0\le p,a_i,c\le 10^9$，$1\le t\le g\le n$。

核心思路

乘法标记是可以影响加法标记的。
而加法标记则不影响乘法标记。
所以要利用交换律。
若ax + b 要乘以k 则 ax * k+b * k = k*(ax+b)。
而加法不影响。
这里指Lazy标记的修改。

AC代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll read()
{
	ll r=0,f=1;char c=getchar();
	while((c<'0'||c>'9')&&c!='-') c=getchar();
	if(c=='-') f=-1,c=getchar();
	while(c>='0'&&c<='9') r=r*10+c-'0',c=getchar();
	return r*f;
}
struct Segment
{
	ll l,r,sum,add,mul;
	#define l(x) tree[x].l
	#define r(x) tree[x].r
	#define sum(x) tree[x].sum
	#define add(x) tree[x].add
	#define mul(x) tree[x].mul
}tree[100010*4];
ll n,m,h,a[100010];
void build(ll p,ll l,ll r)
{
	l(p)=l,r(p)=r,mul(p)=1,add(p)=0;
	if(l==r){sum(p)=a[l];return ;}
	ll mid=(l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	sum(p)=sum(p*2)+sum(p*2+1);
	sum(p)%=h;
}
void spread(ll p)
{
	if(add(p)||mul(p)!=1)
	{
		sum(p*2)=(sum(p*2)*mul(p))%h;
		sum(p*2+1)= (sum(p*2+1)*mul(p))%h;
		sum(p*2)=(sum(p*2)+add(p)*(r(p*2)-l(p*2)+1))%h;
		sum(p*2+1)=(sum(p*2+1)+add(p)*(r(p*2+1)-l(p*2+1)+1))%h;
		mul(p*2)=(mul(p*2)*mul(p))%h;
		mul(p*2+1)=(mul(p*2+1)*mul(p))%h;
		add(p*2)=(add(p*2)*mul(p))%h;
		add(p*2+1)=(add(p*2+1)*mul(p))%h;
		add(p*2)=(add(p*2)+add(p))%h;
		add(p*2+1)=(add(p*2+1)+add(p))%h;
		add(p)=0;
		mul(p)=1;
	}
}
void change(ll p,ll l,ll r,ll d,ll k)
{
	if(l<=l(p)&&r>=r(p))
	{
		if(k==2)
		{
			sum(p)+=d*(r(p)-l(p)+1);
			add(p)+=d;
			sum(p)%=h;
			add(p)%=h;
		}
		if(k==1)
		{
			sum(p)*=d;
			add(p)*=d;
			mul(p)*=d;
			sum(p)%=h;
			add(p)%=h;
			mul(p)%=h;
		}
		return ;
	}
	spread(p);
	ll mid=(l(p)+r(p))>>1;
	if(l<=mid) change(p*2,l,r,d,k);
	if(r>mid) change(p*2+1,l,r,d,k);
	sum(p)=sum(p*2)+sum(p*2+1);
	sum(p)%=h;
}
ll ask(ll p,ll l,ll r)
{
	if(l<=l(p)&&r>=r(p)) return sum(p);
	spread(p);
	ll mid=(l(p)+r(p))>>1;
	ll ans=0;
	if(l<=mid) ans+=ask(p*2,l,r);
	if(r>mid) ans+=ask(p*2+1,l,r);
	ans%=h;
	return ans;
}
int main()
{
    cin>>n>>h;
	for(int i=1;i<=n;i++)cin>>a[i];
	cin>>m;
	build(1,1,n);	
	for(int i=1;i<=m;i++)
	{
		ll q;
		cin>>q;
		if(q==1)
		{
			ll x,y,z;
			cin>>x>>y>>z;
			change(1,x,y,z,1);
		}
		else if(q==2){
			ll x,y,z;
			cin>>x>>y>>z;
			change(1,x,y,z,2);
		}
		else if(q==3){
			ll x,y,z;
			cin>>x>>y;
			printf("%lld\n",ask(1,x,y)%h);
		}
	}
	return 0;
}