线段树学习笔记——1

P3373 【模板】线段树 2 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题意：

        对一个数组有三个操作

• 将某区间每一个数乘上 x；
• 将某区间每一个数加上 x；
• 求出某区间每一个数的和。

        需要一直维护这个区间和，线段树是最优解法，时间复杂度为O(nlogn)

        这题的难点在于乘法操作和加法操作顺序的不确定性，有可能先乘再加，也可能先加再乘......需要思考的是两个懒标记的下传

思路：

        一开始会想到比较暴力的做法，就是如果在当前节点碰到有加法（或者乘法）懒标记的话，就先将懒标记下传，这样能确保这两个懒标记不会重复，但是这样会re、、、、、递归太多次啦

        因此，我们可以写一下数学式子，就是有可能是(sum + add) * mul，也可能是sum * mul + add，前者分配一下得到 sum * mul + add * mul，不难发现，区别是加数乘上了乘数，这样我们可以一直维护 sum * mul + add这个区间和，遇到乘法的话需要将add懒标记乘上乘数。

        下传懒标记的时候，就是要先下传乘法懒标记，再下传加法懒标记，这样就有个明确的顺序

代码：

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 1e5+10;
int a[N],w[4*N],m;
int lzy_add[4*N],lzy_mul[4*N];
void up(int u)
{
	w[u] = (w[u*2] + w[u*2+1]) % m;
}
void build(int u,int L,int R) // 建树
{
	if (L == R)
	{
		w[u] = a[L];
		return;
	}
	lzy_mul[u] = 1;
	int M = L+R >> 1;
	build(u*2,L,M);
	build(u*2+1,M+1,R);
	up(u);
}
bool in(int L,int R,int l,int r)
{
	return (l <= L) && (R <= r);
}
bool out(int L,int R,int l,int r)
{
	return (L > r) || (l > R);
}
void down(int u,int L,int R);
void marktag(int u,int L,int R,int x,int type) // 打懒标记
{
	if (type == 1)
	{
		lzy_mul[u] = (lzy_mul[u] * x) % m;
		lzy_add[u] = (lzy_add[u] * x) % m; // + add * mul
		w[u] = w[u]*x % m;
	}
	if (type == 2)
	{
		lzy_add[u] = (lzy_add[u] + x) % m;
		w[u] += (R-L+1)*x % m;
	}
}
void down(int u,int L,int R) // 下传
{
	int M = L+R >> 1;
    //先下传乘法标记
	marktag(u*2,L,M,lzy_mul[u],1);
	marktag(u*2+1,M+1,R,lzy_mul[u],1);
	lzy_mul[u] = 1;
	
    //再下传加法标记
	marktag(u*2,L,M,lzy_add[u],2);
	marktag(u*2+1,M+1,R,lzy_add[u],2);
	lzy_add[u] = 0;
}
int query(int u,int L,int R,int l,int r)
{
	if (in(L,R,l,r)) return w[u];
	else if (!out(L,R,l,r))
	{
		int M = L+R >> 1;
		down(u,L,R);
		return query(u*2,L,M,l,r) + query(u*2+1,M+1,R,l,r);
	}
	else return 0;
}
void update(int u,int L,int R,int l,int r,int x,int type)
{
	if (in(L,R,l,r)) marktag(u,L,R,x,type);
	else if (!out(L,R,l,r))
	{
		int M = L+R >> 1;
		down(u,L,R);
		update(u*2,L,M,l,r,x,type);
		update(u*2+1,M+1,R,l,r,x,type);
		up(u);
	}
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	int n,q,x,y,k,op;
	cin >> n >> q >> m;
	for (int i = 1; i <= n; ++i) cin >> a[i];
	build(1,1,n);
	for (int i = 1; i <= q; ++i)
	{
		cin >> op;
		if (op == 1)
		{
			cin >> x >> y >> k;
			update(1,1,n,x,y,k,1);
		}
		else if (op == 2)
		{
			cin >> x >> y >> k;
			update(1,1,n,x,y,k,2);
		}
		else if (op == 3)
		{
			cin >> x >> y;
			cout << query(1,1,n,x,y) % m << endl;
		}
	}
	return 0;
}