[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。
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define L u<<1
#define R u<<1|1
const int N=1e5+5;
int n,mod,q,a[N];
struct node{
int l,r,v;
int add,mul;
}tr[N*4];
void pushup(int u){
tr[u].v=(tr[L].v+tr[R].v)%mod;
}
void pushdown(int u){
tr[L].v=(tr[L].v*tr[u].mul+tr[u].add*(tr[L].r-tr[L].l+1))%mod;
tr[R].v=(tr[R].v*tr[u].mul+tr[u].add*(tr[R].r-tr[R].l+1))%mod;
tr[L].add=(tr[L].add*tr[u].mul+tr[u].add)%mod;
tr[R].add=(tr[R].add*tr[u].mul+tr[u].add)%mod;
tr[L].mul=(tr[L].mul*tr[u].mul)%mod;
tr[R].mul=(tr[R].mul*tr[u].mul)%mod;
tr[u].add=0,tr[u].mul=1;
}
void build(int u,int l,int r){
tr[u].l=l,tr[u].r=r;
tr[u].mul=1,tr[u].add=0;
if(l==r){
tr[u].v=a[l]%mod;
return;
}
int mid=l+r>>1;
build(L,l,mid),build(R,mid+1,r);
pushup(u);
}
void modifyMul(int u,int l,int r,int k){
if(tr[u].l>=l&&tr[u].r<=r){
tr[u].v=(tr[u].v*k)%mod;
tr[u].mul=(tr[u].mul*k)%mod;
tr[u].add=(tr[u].add*k)%mod;
return;
}
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid) modifyMul(L,l,r,k);
if(r>mid) modifyMul(R,l,r,k);
pushup(u);
}
void modifyAdd(int u,int l,int r,int k){
if(tr[u].l>=l&&tr[u].r<=r){
tr[u].v=(tr[u].v+k*(tr[u].r-tr[u].l+1))%mod;
tr[u].add=(tr[u].add+k)%mod;
return;
}
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid) modifyAdd(L,l,r,k);
if(r>mid) modifyAdd(R,l,r,k);
pushup(u);
}
int query(int u,int l,int r){
if(tr[u].l>=l&&tr[u].r<=r) return tr[u].v;
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
int ans=0;
if(l<=mid) ans=(ans+query(L,l,r))%mod;
if(r>mid) ans=(ans+query(R,l,r))%mod;
return ans%mod;
}
void solve(){
scanf("%lld%lld",&n,&mod);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
build(1,1,n);
scanf("%lld",&q);
while(q--){
int op,x,y,k;
scanf("%lld",&op);
if(op==1){
scanf("%lld%lld%lld",&x,&y,&k);
modifyMul(1,x,y,k);
}
else if(op==2){
scanf("%lld%lld%lld",&x,&y,&k);
modifyAdd(1,x,y,k);
}
else{
scanf("%lld%lld",&x,&y);
printf("%lld\n",query(1,x,y));
}
}
}
signed main(){
int t=1;
// scanf("%d",&t);
while(t--) solve();
return 0;
}