线段树模板题

P3372 【模板】线段树 1

#include <iostream>
#include <algorithm>
typedef long long LL;
using namespace std;
const int MAXN = 1000000+5;
LL arr[MAXN];
struct seg { int l, r, lz; LL v; }t[MAXN << 2];//lz延迟标记,l 左边界,r 右边界
int lch(int k) { return k << 1; };
int rch(int k) { return k << 1 | 1; };
inline void push_down(int k) {
	if (!t[k].lz) return;
	int mid = (t[k].l + t[k].r) >> 1, lz = t[k].lz;
	t[lch(k)].lz += lz;
	t[rch(k)].lz += lz;
	t[lch(k)].v += lz * (mid - t[lch(k)].l + 1);
	t[rch(k)].v += lz * (t[rch(k)].r - mid);
	t[k].lz = 0;
}
inline void push_up(int k) { t[k].v = t[lch(k)].v + t[rch(k)].v; }
void build(int k, int l, int r) {//建树
	t[k].l = l; t[k].r = r; t[k].lz = 0;
	if (l == r) {//叶节点
		t[k].v = arr[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(lch(k),l,mid);
	build(rch(k),mid+1,r);
	push_up(k);
}
LL query(int k,int l,int r) {//查询区间[l,r]的和
	if (t[k].l >= l && t[k].r <= r) return t[k].v;
	LL s = 0;
	push_down(k);
	if (t[lch(k)].r >= l) s+=query(lch(k),l,r);
	if (t[rch(k)].l <= r) s+=query(rch(k),l,r);
	return s;
}
void updata(int k, int l,int r,int v) {//区间[l,r]加上v
	if (t[k].l >= l && t[k].r <= r) {
		t[k].v += v * (t[k].r - t[k].l + 1);
		t[k].lz += v;
		return;
	}
	push_down(k);
	if (t[lch(k)].r >= l) updata(lch(k), l, r, v);
	if (t[rch(k)].l <= r) updata(rch(k), l, r, v);
	push_up(k);
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int N, M, a, b, c, d;
	cin >> N >> M;
	for (int i = 1; i <= N; i++) {
		cin >> arr[i];//输入数列
	}
	build(1, 1, N);
	for (int i = 0; i < M; i++) {
		cin >> a;
		if (a == 1) {//修改操作
			cin >> b >> c >> d;
			updata(1, b, c, d);
		}
		else {//查询操作
			cin >> b >> c;
			cout << query(1, b, c) << endl;
		}
	}
	return 0;
}