模板——线段树 区间更新

线段树区间更新_ACM模板

题目描述
已知一个数列
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。
接下来M行每行包含3或4个整数，表示一个操作，具体如下：
操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k；
操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和？

typedef long long ll;
const int maxn = 1e5 + 10;//N的上限
int a[maxn];//a[i]：id为i的数据
struct node {
	int l, r;//节点的左界、右界
	ll sum, lazy;//节点的数据、lazy标记
	void update(ll val) {//更新节点的数据、lazy值
		sum += (r - l + 1)*val;
		lazy += val;
	}
}s[4 * maxn];
void p_up(int k) {//更新k节点存储的值
	s[k].sum = s[2 * k].sum + s[2 * k + 1].sum;
	return;
}
void p_down(int k) {//更新k节点的lazy值，将lazy传递给孩子节点
	ll val = s[k].lazy;
	if (val) {
		s[2 * k].update(val);
		s[2 * k + 1].update(val);
		s[k].lazy = 0;
	}
}
void bulid(int l, int r, int k) {//创建线段树
	s[k].l = l;
	s[k].r = r;
	s[k].lazy = 0;
	if (l == r) {
		s[k].sum = a[l];
		return;
	}
	int mid = (l + r) / 2;
	bulid(l, mid, 2 * k);
	bulid(mid + 1, r, 2 * k + 1);
	p_up(k);
}
void update(int l, int r, int k, ll val) {
//将[l,r]区间的每个数都加上val，更新线段树
	if (l <= s[k].l && s[k].r <= r) {
		s[k].update(val);
		return;
	}
	p_down(k);
	int mid = (s[k].l + s[k].r) / 2;
	if (l <= mid) update(l, r, 2 * k, val);
	if (r > mid) update(l, r, 2 * k + 1, val);
	p_up(k);
}
ll fin(int l, int r, int k) {//返回[l,r]区间的和
	if (l <= s[k].l && s[k].r <= r) {
		return s[k].sum;
	}
	ll sum = 0;
	p_down(k);
	int mid = (s[k].l + s[k].r) / 2;
	if (l <= mid) sum += fin(l, r, 2 * k);
	if (r > mid) sum += fin(l, r, 2 * k + 1);
	p_up(k);
	return sum;
}