洛谷P2023 [AHOI2009]维护序列

P2023 [AHOI2009]维护序列

题目链接-P2023 [AHOI2009]维护序列



解题思路
经典线段树题目

• 因为运算包括乘法和加法，这时候就要考虑优先的问题;
• 当对某区间执行加法操作时，由于加法优先级低，不会对乘法操作产生影响，故直接相加即可；
• 当对某区间执行乘法操作时，由于乘法优先级高，会对之前的加法操作产生影响，故需要在相乘时不仅对sum和mul相乘，也需要对plu相乘；
• 综上，所以要先乘后加，即所有操作统一为对该节点*mult +add的形式
• 对区间进行修改时void updata(ill l,ll r,ll add,ll mult,int i)，那么乘法操作就成了updata(l,r,0,v,1)，加法操作就是updata(l,r,v,1,1);
• pushdown()函数维护lazytag时还是照样先乘后加的顺序维护，然后，lazy标记叠加时候，要把顺序考虑进去
• sum*a+b a和b这是儿子原来的lazytag 然后加入c和d是父亲的lazytag，假设x和y是维护后新的lazytag,那么按照先乘后加应该这么算:(sum * a+b) * c+d=sum * x+y ；
  化简得 sum * a * c+b * c+d=sum * x+y；
  最后解得x = a * c ，y = b * c+d ，然后就可以更新了
• 其他基本是按照线段树的板子，具体操作见代码

注意
一定记得要取模！！！而且建树时要把乘法lazytag(即t[i].mul)置为1！！！如果不用读入优化就老老实实scanf，否则容易超时

做这个题之前或者做完这个题想再巩固一下,可以看看这道题，两道题几乎一模一样——>洛谷P3373 【模板】线段树 2

附上代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<iomanip>
#include<cassert>
#include<climits>
#include<list>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<unordered_map>
#include<unordered_set>
#define lowbit(x) (x &(-x))
using namespace std;
const int INF=0x3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=1e5+5;
typedef long long ll;
typedef pair<int,int> PII;
ll a[N],n,p;
struct node{
	ll l,r;
	ll mul,plu,sum;
}t[N<<2];//线段树结构体，sum表示此时的答案，mul表示乘法上的lazy标记，plu是加法上的
void push_up(ll i){
	t[i].sum=(t[i<<1].sum+t[i<<1|1].sum)%p;
}
void build(ll l,ll r,ll i){
//i为当前需要建立的结点，l为当前需要建立区间的左端点，r则为右端点
	t[i].l=l;t[i].r=r;t[i].mul=1;
	if(l==r){//左端点等于右端点，即为叶子节点，直接赋值即可
		t[i].sum=a[l]%p;
		return ;
	}
	ll m=l+((r-l)>>1);//m则为中间点，左儿子的结点区间为[l,m],右儿子的结点区间为[m+1,r]
	build(l,m,i<<1);
	build(m+1,r,i<<1|1);//构造左右子节点
	push_up(i); //更新父节点
}
void pushdown(ll i){//维护lazytag
	t[i<<1].mul=(t[i<<1].mul*t[i].mul)%p;
    t[i<<1].plu=((t[i<<1].plu*t[i].mul)%p+t[i].plu%p)%p;
    t[i<<1].sum=((t[i<<1].sum*t[i].mul)%p+(t[i].plu*(t[i<<1].r-t[i<<1].l+1))%p)%p;

    t[i<<1|1].mul=(t[i<<1|1].mul*t[i].mul)%p;
    t[i<<1|1].plu=((t[i<<1|1].plu*t[i].mul)%p+t[i].plu%p)%p;
    t[i<<1|1].sum=((t[i<<1|1].sum*t[i].mul)%p+(t[i].plu*(t[i<<1|1].r-t[i<<1|1].l+1))%p)%p;

	t[i].mul=1;
	t[i].plu=0;
}
void updata(ll l,ll r,ll add,ll mult,ll i){
//l,r为修改的区间范围，add为加的数，mult为乘的数，i为结点下标
	if(l<=t[i].l&&t[i].r<=r){
		t[i].mul=(t[i].mul*mult)%p;
		t[i].plu=((t[i].plu*mult)%p+add%p)%p;
		t[i].sum=((t[i].sum*mult)%p+add*(t[i].r-t[i].l+1)%p)%p;
		return ;
	}
	pushdown(i);
	ll m=t[i].l+((t[i].r-t[i].l)>>1);
	if(l<=m)
		updata(l,r,add,mult,i<<1);
	if(m<r)
		updata(l,r,add,mult,i<<1|1);
	push_up(i);//更新父节点
}
ll query(ll l,ll r,ll i){
	if(l<=t[i].l&&t[i].r<=r)
	//如果当前结点的区间真包含于要查询的区间内，则返回结点信息且不需要往下递归
		return t[i].sum;
	pushdown(i);//下放lazytag
	ll res=0;
	ll m=t[i].l+((t[i].r-t[i].l)>>1);
	if(l<=m)//如果左子树和需要查询的区间交集非空
		res+=query(l,r,i<<1);
	if(m<r)//如果右子树和需要查询的区间交集非空，不能用else,因为查询区间可能同时和左右区间都有交集
		res+=query(l,r,i<<1|1);
	return res; 
}
signed main(){
	
	ll m;
	scanf("%lld%lld",&n,&p);
	for(ll i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	build(1,n,1);
	scanf("%lld",&m);
	while(m--){
		ll op,l,r,c;
		scanf("%lld%lld%lld",&op,&l,&r);
		if(op==1){
			scanf("%lld",&c);
			updata(l,r,0,c,1);
		}
		else if(op==2){
			scanf("%lld",&c);
			updata(l,r,c,1,1);
		}
		else
			printf("%lld\n",query(l,r,1)%p);
	}
	return 0;
}