7.30日结

修改中含有加和乘的线段树
传送门

#include <bits/stdc++.h>
#define MAXN 100010
#define ll long long
using namespace std;
int n, m, mod;
int a[MAXN];
struct node{
	ll sum, add, mul;
	int l, r;
}s[MAXN*4];
void pushdown(int pos) {
	s[pos << 1].sum = (s[pos << 1].sum * s[pos].mul + s[pos].add * (s[pos << 1].r - s[pos << 1].l + 1)) % mod;
	s[pos << 1 | 1].sum = (s[pos << 1 | 1].sum * s[pos].mul + s[pos].add * (s[pos << 1 | 1].r - s[pos << 1 | 1].l + 1)) % mod;
	s[pos << 1].mul = (s[pos << 1].mul * s[pos].mul) % mod;
	s[pos << 1 | 1].mul = (s[pos << 1 | 1].mul * s[pos].mul) % mod;
	s[pos << 1].add = (s[pos << 1].add * s[pos].mul + s[pos].add) % mod;
	s[pos << 1 | 1].add = (s[pos << 1 | 1].add * s[pos].mul + s[pos].add) % mod;
	s[pos].add = 0;
	s[pos].mul = 1;
	return;
}
void build(int pos, int l, int r) {
	s[pos].l = l;
	s[pos].r = r;
	s[pos].mul = 1;
	if (l == r) {
		s[pos].sum = a[l] % mod;
		return;
	}
	int mid = (l + r) >> 1;
	build(pos << 1, l, mid);
	build(pos << 1 | 1, mid + 1, r);
	s[pos].sum = (s[pos << 1].sum + s[pos <<1 | 1].sum) % mod;
	return;
}
void mul(int pos, int x, int y, int k) {
	if (x <= s[pos].l && s[pos].r <= y) {
		s[pos].add = (s[pos].add * k) % mod;
		s[pos].mul = (s[pos].mul * k) % mod;
		s[pos].sum = (s[pos].sum * k) % mod;
		return;
	}
	pushdown(pos);
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) mul(pos << 1, x, y, k);
	if (y > mid) mul(pos << 1 | 1, x, y, k);
	s[pos].sum = (s[pos << 1].sum + s[pos << 1 | 1].sum) % mod;
	return;
}
void add(int pos, int x, int y, int k) {
	if (x <= s[pos].l && s[pos].r <= y) {
		s[pos].add = (s[pos].add + k) % mod;
		s[pos].sum = (s[pos].sum + k * (s[pos].r - s[pos].l + 1)) % mod;
		return;
	}
	pushdown(pos);
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) add(pos << 1, x, y, k);
	if (y > mid) add(pos << 1 | 1, x, y, k);
	s[pos].sum = (s[pos << 1].sum + s[pos << 1 | 1].sum) % mod;
	return;
}
ll AskRange(int pos, int x, int y) { //区间询问
	if (x <= s[pos].l && s[pos].r <= y) {
		return s[pos].sum;
	}
	pushdown(pos);
	ll val = 0;
	int mid = (s[pos].l + s[pos].r) >> 1;
	if (x <= mid) val = (val + AskRange(pos << 1, x, y)) % mod;
	if (y > mid) val = (val + AskRange(pos << 1 | 1, x, y)) % mod;
	return val;
}
int main() {
	scanf("%d%d%d", &n, &m, &mod);
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	build(1, 1, n);
	for (int i = 1; i <= m; i++) {
		int opt, x, y;
		scanf("%d%d%d", &opt, &x, &y);
		if (opt == 1) {
			int k;
			scanf("%d", &k);
			mul(1, x, y, k);
		}
		if (opt == 2) {
			int k;
			scanf("%d", &k);
			add(1, x, y, k);
		}
		if (opt == 3) {
			printf("%lld\n", AskRange(1, x, y));
		}
	}

	return 0;
}

n << 1 == n * 2,n << 1 | 1 == n * 2 + 1
结构体的大小要乘多倍,不然会runtime
结构体内变量名和自定义函数名可以重复,不在一个作用域 ,但是最好不要

下面这段先忽略,我还没想好怎么写,再改改。
“先乘后加”的意思是先更新lazy*,再更新lazy+(同时还要乘lazy*),公式为:

t[i].sum=t[i].sum*t[i].(lazy*+t[i].(lazy+)*(t[i].r-t[i].l+1)

以对某区间两种操作为例,
1.先加n再乘m,原值x,结果应该是xm+nm,先加n,lazy* =1,lazy+ =n,再乘m,lazy* =m,lazy+ =mn,
2.先乘m再加n,原值x,结果应该是x
m+n,先乘m,

©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页