P2023 [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$。

#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;
}