线段树超详细笔记4

第四道题，我们来看看若是需要维护多个懒标记的情形~

[AHOI2009] 维护序列 luogu链接：https://www.luogu.com.cn/problem/P2023（写懒标记变懒了，懒得粘贴题目了）

这是洛谷有名的“懒标记运算优先级”问题

既要区间乘，又要区间加，其中一个询问倒是蛮简单的，也就是询问区间和

之前我们做过的题目是只有区间加法，也就是懒标记的基础应用，倒是蛮简单的，现在我们要同时做区间乘法和区间加法

很容易想到，再加上一个维护区间乘法的懒标记multi呗

但是你有没有想过我们在modify函数和query函数里如何更新这两个懒标记吗？

之前的画风是这样的~

void pushdown(int u)
{
	node &root = tr[u],&l = tr[u<<1],&r = tr[u<<1|1];
	if(root.all)
	{
		l.all += root.all,l.sum += (ll)(l.r-l.l+1)*root.all;
		r.all += root.all,r.sum += (ll)(r.r-r.l+1)*root.all;
		root.all = 0;
	}
}

现在多了一个标记，那么我们是在加法后面搞乘法吗？

我们需要注意一点：在做pushdown操作的时候两个懒标记可能是同时存在的，但是我们并不清楚实际在modify的时候是加法在前还是乘法在前

若是我们在pushdown中规定了一种顺序，那我们在处理两种懒标记的时候就只会一直选择一种情况——先处理加法，或者先处理乘法

若是我们选择加法在前，我们要考虑的问题是若是乘法的懒标记先被加上，那么我们先处理加法是否妥当

我们不妨设加法的懒标记为add，乘法的懒标记是multi，我们若是先做加法，那么我们在统计结果的时候实际加上的值应当是add/multi——

怎么能搞除法嘞？万一有小数嘞？万一除不尽嘞？

所以我们理应考虑第二种情况，也就是先算乘法，也就是考虑若是加法的懒标记先被加上，我们先算乘法是否合理

然后我们惊奇的发现，先算乘法就好使的多！

即使是加法的懒标记先被加上，我们先算的乘法，那么我们再在后面补一个multi*add就可

由此我们形成了一种套路，也就是先乘再加，实际上就是把两个懒标记搞成相当于只有一个懒标记的形式罢了

是的，只要保证在pushdown之后我们复合而成的懒标记形式不发生变化（这道题的形式是a*b+c）我们就可以把两个懒标记复合！

我们直接上代码！

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N = 100010;
int n,p,m;
ll w[N];
struct node
{
	int l,r;
	ll sum,add,mul;
}tr[N*4];

void pushup(int u)
{
	tr[u].sum = (tr[u<<1].sum+tr[u<<1|1].sum)%p;
}

void eval(node &root,int add,int mul)
{
	root.sum = ((ll)(root.sum*mul)+(ll)(root.r-root.l+1)*add)%p;
	root.mul = (ll)root.mul*mul%p;
	root.add = (ll)(root.add*mul+add)%p;
}

void pushdown(int u)
{
	eval(tr[u<<1],tr[u].add,tr[u].mul);
	eval(tr[u<<1|1],tr[u].add,tr[u].mul);
	tr[u].add = 0,tr[u].mul = 1;
}

void build(int u,int l,int r)
{
	if(l==r) tr[u] = {l,r,w[r],0,1};
	else
	{
		tr[u] = {l,r,0,0,1};
		int mid = l+r>>1;
		build(u<<1,l,mid);build(u<<1|1,mid+1,r);
		pushup(u);
	}
}

void modify(int u,int l,int r,int add,int mul)
{
	if(tr[u].l>=l && tr[u].r<=r) eval(tr[u], add, mul);
	else
	{
		pushdown(u);
		int mid = tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1,l,r,add,mul);
		if(r>mid) modify(u<<1|1,l,r,add,mul);           
		pushup(u);
	} 
}

int query(int u,int l,int r)
{
	if(tr[u].l>=l && tr[u].r<=r) return tr[u].sum;

	pushdown(u);
	int mid = tr[u].l+tr[u].r>>1;
	int sum = 0;
	if(l<=mid) sum = sum + query(u<<1,l,r)%p;
	if(r>mid) sum = (sum+query(u<<1|1,l,r))%p;
	return sum;
}

int main()
{
	cin>>n>>p;
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	build(1,1,n);
	
	cin>>m;
	while(m--)
	{
		int t,l,r,d;
		cin>>t>>l>>r;
		if(t==1)
		{
			cin>>d;
			modify(1,l,r,0,d);
		}
		else if(t==2)
		{
			cin>>d;
			modify(1,l,r,d,1);
		}
		else cout<<query(1,l,r)<<endl;
	}
	return 0;
} 

啊这不得加油把线段树干的透透的？！