P3372【模板】线段树 1 - 线段树入门练习

版权声明：原创文章转载时请注明出处 https://blog.csdn.net/weixin_42856843/article/details/89054014

P3372【模板】线段树 1 - 线段树入门练习

题目链接：https://www.luogu.org/problemnew/show/P3372

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入输出格式
输入格式：
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式：
输出包含若干行整数，即为所有操作2的结果。

输入输出样例

输入样例#1：

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出样例#1：

11
8
20

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_，保证在int64/long long数据范围内）

样例说明：

思路

没什么思路，套模板就行，注意数据规模，把树开大些

代码

#include <iostream>
#include <algorithm>
#include <string.h>
#define maxn 1000005
#define inf 0x3f3f3f3f3f3f3f3f
#define ll long long
#define _for(i, a) for(ll i = 0; i < (a); i++)
#define _rep(i, a, b) for(ll i = (a); i <= (b); i++)
using namespace std;
ll arr[maxn], SegTree[maxn * 4], ad[maxn * 4];
void build(ll node, ll begin, ll end) {
	if (begin == end) {
		SegTree[node] = arr[begin];
	}
	else {
		ll mid = (begin + end) / 2;
		build(2 * node, begin, mid);
		build(2 * node + 1, mid + 1, end);
		SegTree[node] = SegTree[2 * node] + SegTree[2 * node + 1];
	}
}
void add(ll node, ll begin, ll end, ll left, ll right, ll val) {
	if (begin == left && end == right) {
		ad[node] += val;
		return;
	}
	SegTree[node] += (right - left + 1) * val;
	ll mid = (begin + end) / 2;
	if (mid >= right) add(node * 2, begin, mid, left, right, val);
	else if (mid < left) add(node * 2 + 1, mid + 1, end, left, right, val);
	else {
		add(node * 2, begin, mid, left, mid, val);
		add(node * 2 + 1, mid + 1, end, mid + 1, right, val);
	}
}
ll Getsum(ll node, ll begin, ll end, ll left, ll right) {
	if (ad[node]) {
		ad[node * 2] += ad[node];
		ad[node * 2 + 1] += ad[node];
		SegTree[node] += ad[node] * (end - begin + 1);
		ad[node] = 0;
	}
	ll mid = (begin + end) / 2;
	if (begin == left && end == right)
		return SegTree[node];
	else if (mid >= right)
		return Getsum(node * 2, begin, mid, left, right);
	else if (mid < left)
		return Getsum(node * 2 + 1, mid + 1, end, left, right);
	else
		return Getsum(node * 2, begin, mid, left, mid) + Getsum(node * 2 + 1, mid + 1, end, mid + 1, right);
}
int main() {
	//freopen("in.txt", "r", stdin);
	ios::sync_with_stdio(false);
	memset(ad, 0, sizeof(ad));
	ll N, M, c, l, r, val;
	cin >> N >> M;
	_for(i, N) cin >> arr[i];
	build(1, 0, N - 1);
	_for(i, M) {
		cin >> c;
		if (c == 1) {
			cin >> l >> r >> val;
			add(1, 0, N - 1, l - 1, r - 1, val);
		}
		else {
			cin >> l >> r;
			cout << Getsum(1, 0, N - 1, l - 1, r - 1) << "\n";
		}
	}
	return 0;
}