[AHOI2009] 维护序列
题目背景
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。
题目描述
有一个长为 n n n 的数列 { a n } \{a_n\} {an},有如下三种操作形式:
- 格式
1 t g c
,表示把所有满足 t ≤ i ≤ g t\le i\le g t≤i≤g 的 a i a_i ai 改为 a i × c a_i\times c ai×c ; - 格式
2 t g c
表示把所有满足 t ≤ i ≤ g t\le i\le g t≤i≤g 的 a i a_i ai 改为 a i + c a_i+c ai+c ; - 格式
3 t g
询问所有满足 t ≤ i ≤ g t\le i\le g t≤i≤g 的 a i a_i ai 的和,由于答案可能很大,你只需输出这个数模 p p p 的值。
输入格式
第一行两个整数 n n n 和 p p p。
第二行含有 n n n 个非负整数,表示数列 { a i } \{a_i\} {ai} 。
第三行有一个整数 m m 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 1 次操作后,数列为 { 1 , 10 , 15 , 20 , 25 , 6 , 7 } \{1,10,15,20,25,6,7\} {1,10,15,20,25,6,7}。
- 对第 2 2 2 次操作,和为 10 + 15 + 20 = 45 10+15+20=45 10+15+20=45,模 43 43 43 的结果是 2 2 2。
- 经过第 3 3 3 次操作后,数列为 { 1 , 10 , 24 , 29 , 34 , 15 , 16 } \{1,10,24,29,34,15,16\} {1,10,24,29,34,15,16}。
- 对第 4 4 4 次操作,和为 1 + 10 + 24 = 35 1+10+24=35 1+10+24=35,模 43 43 43 的结果是 35 35 35。
- 对第 5 5 5 次操作,和为 29 + 34 + 15 + 16 = 94 29+34+15+16=94 29+34+15+16=94,模 43 43 43 的结果是 8 8 8。
数据规模与约定
测试数据规模如下表所示:
数据点编号 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9,10 |
---|---|---|---|---|---|---|---|---|---|
n = n= n= | 10 10 10 | 1000 1000 1000 | 1000 1000 1000 | 10000 10000 10000 | 60000 60000 60000 | 70000 70000 70000 | 80000 80000 80000 | 90000 90000 90000 | 100000 100000 100000 |
m = m= m= | 10 10 10 | 1000 1000 1000 | 1000 1000 1000 | 10000 10000 10000 | 60000 60000 60000 | 70000 70000 70000 | 80000 80000 80000 | 90000 90000 90000 | 100000 100000 100000 |
对于全部的测试点,保证 0 ≤ p , a i , c ≤ 1 0 9 0 \leq p, a_i, c \leq 10^9 0≤p,ai,c≤109, 1 ≤ t ≤ g ≤ n 1 \leq t \leq g \leq n 1≤t≤g≤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;
}