[bzoj5039][Jsoi2014]序列维护 线段树

5039: [Jsoi2014]序列维护

Time Limit: 20 Sec   Memory Limit: 256 MB
[ Submit][ Status][ Discuss]

Description

JYY 有一个维护数列的任务。 他希望你能够来帮助他完成。

JYY 现在有一个长度为 N 的序列 a1,a2,…,aN，有如下三种操作：

1、 把数列中的一段数全部乘以一个值；

2、 把数列中的一段数全部加上一个值；

3、 询问序列中的一段数的和。

由于答案可能很大，对于每个询问，你只需要告诉 JYY 这个询问的答案对 P

取模的结果即可。

Input

第一行包含两个正整数， N 和 P；

第二行包含 N 个非负整数，从左到右依次为 a1,a2,…,aN。

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

接下来 M 行，每行满足如下三种形式之一：

1、“ 1 t g c”（不含引号）。表示把所有满足 t ≤ i ≤ g 的 ai 全部乘以 c；

2、“ 2 t g c”（不含引号）。表示把所有满足 t ≤ i ≤ g 的 ai 全部加上 c；

3、“ 3 t g”（不含引号）。表示询问满足 t ≤ i ≤ g 的 ai 的和对 P 取模的值。

1 ≤ N,M ≤ 10^5， 1 ≤ P, c, ai ≤ 2*10^9， 1 ≤ t ≤ g ≤ N

Output

对于每个以 3 开头的操作，依次输出一行，包含对应的结果。

Sample Input

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

Sample Output

2
35
8
【样例说明】
初始时数列为(1,2,3,4,5,6,7)。
经过第 1 次操作后，数列为(1,10,15,20,25,6,7)。
对第 2 次操作，和为 10+15+20=45，模 43 的结果是 2。
经过第 3 次操作后，数列为(1,10,24,29,34,15,16}
对第 4 次操作，和为 1+10+24=35，模 43 的结果是 35。
对第 5 次操作，和为 29+34+15+16=94,模 43 的结果是 8。

HINT

重点是pushdown

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1000000 + 5;
long long p,sum[N],f1[N],f2[N],a[100005];
void pushdown( int k, int l, int r ){
	int mid = l + r >> 1;
	if( f1[k] == 0 && f2[k] == 1 ) return ;
	f1[k<<1] = ( f1[k<<1] * f2[k] % p + f1[k] ) % p;
	f2[k<<1] = f2[k<<1] * f2[k] % p;
	sum[k<<1] = ( ( sum[k<<1] * f2[k] % p ) + ( f1[k] * (LL)(mid-l+1) ) % p ) % p;
	f1[k<<1|1] = ( ( f1[k<<1|1] * f2[k] ) % p + f1[k] ) % p;
	f2[k<<1|1] = f2[k<<1|1] * f2[k] % p;
	sum[k<<1|1] = ( ( sum[k<<1|1] * f2[k] ) % p + ( f1[k] * (LL)(r-mid) ) % p) % p;
	f1[k] = 0; f2[k] = 1;
}
void update( int k ){ sum[k] = ( sum[k<<1] + sum[k<<1|1] ) % p; }
void modify( int k, int l, int r, int L, int R, int val ){
	if( l >= L && r <= R ){
		f1[k] += val; sum[k] += ( val * (LL)( r - l + 1 ) ) % p; sum[k] %= p; return ;
	}
	pushdown( k, l, r );
	int mid = l + r >> 1;
	if( L <= mid ) modify( k<<1, l, mid, L, R, val );
	if( R > mid ) modify( k<<1|1, mid+1, r, L, R, val );
	update( k );
}
void change( int k, int l, int r, int L, int R, int val ){
	if( l >= L && r <= R ){
		f2[k] *= val; f1[k] *= val; f2[k] %= p; f1[k] %= p; sum[k] *= val; sum[k] %= p; return ;
	}
	pushdown( k, l, r );
	int mid = l + r >> 1;
	if( L <= mid ) change( k<<1, l, mid, L, R, val );
	if( R > mid ) change( k<<1|1, mid+1, r, L, R, val );
	update( k );
}
long long query( int k, int l, int r, int L, int R ){
	if( l >= L && r <= R ){
		return sum[k]%p;
	}
	pushdown( k, l, r );
	int mid = l + r >> 1; long long res = 0;
	if( L <= mid ) res = (res+query( k<<1, l, mid, L, R ))%p;
	if( R > mid ) res = (res+query( k<<1|1, mid+1, r, L, R ))%p;
	update( k ); return res;
}
int n,q;
void build( int k, int l, int r ){
	f1[k] = 0; f2[k] = 1;
	if( l == r ){
		sum[k] = a[l]%p; return ;
	}
	int mid = l + r >> 1;
	build( k<<1, l, mid );
	build( k<<1|1, mid+1, r );
	update( k );
}
int main(){
	cin>>n>>p;
	for( int i = 1; i <= n; i++ ) scanf("%d", &a[i]);
	build( 1, 1, n );
	scanf("%d", &q);
	while( q-- ){
		int f,t,g,c;
		scanf("%d%d%d", &f, &t, &g);
		if( f == 1 ){
			scanf("%d", &c);
			change( 1, 1, n, t, g, c );
		}
		if( f == 2 ){
			scanf("%d", &c);
			modify( 1, 1, n, t, g, c );
		}
		if( f == 3 ){
			printf("%d\n", query( 1, 1, n, t, g ));
		}
	}
	return 0;
}