线段树板子写挂了求调!
查看原帖
线段树板子写挂了求调!
572831
WILDFIRE233楼主2023/10/9 22:50

求调!谢谢!

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n, m;
vector<ll> arr, t, lz;
inline void create(int idx, int l, int r) {
	if (l == r) {
		t[idx] = arr[l];
		return;
	}
	int m = l + r >> 1;
	create(idx * 2, l, m), create(idx * 2 + 1, m + 1, r);
	t[idx] = t[idx * 2] + t[idx * 2 + 1];
}
inline void pushDown(int idx, int l, int r) {
	int m = l + r >> 1;
	t[idx * 2] += lz[idx] * (m - l + 1), t[idx * 2 + 1] += lz[idx * 2 + 1] * (r - m);
	lz[idx * 2] += lz[idx], lz[idx * 2 + 1] += lz[idx];
	lz[idx] = 0;
}
inline void update(int idx, int ql, int qr, int l, int r, ll c) {
	if (l >= ql && r <= qr) {
		t[idx] += c * (r - l + 1);
		lz[idx] += c;
		return;
	}
	int m = l + r >> 1;
	if (lz[idx] && l != r) pushDown(idx, l, r);
	if (ql <= m) update(idx * 2, ql, qr, l, m, c);
	if (qr > m) update(idx * 2 + 1, ql, qr, m + 1, r, c);
	t[idx] = t[idx * 2] + t[idx * 2 + 1];
}
inline long long getSum(int idx, int ql, int qr, int l, int r) {
	if (l >= ql && r <= qr) return t[idx];
	int m = l + r >> 1;
	if (lz[idx]) pushDown(idx, l, r);
	ll sum = 0;
	if (ql <= m) sum += getSum(idx * 2, ql, qr, l, m);
	if (qr > m) sum += getSum(idx * 2 + 1, ql, qr, m + 1, r);
	return sum;
}
int main() {
	scanf("%d %d", &n, &m);
	arr.resize(n + 1), t.resize(4 * n), lz.resize(4 * n);
	for (int i = 1; i <= n; i++)
		scanf("%lld", &arr[i]);
	create(1, 1, n);
	for (int i = 0; i < m; i++) {
		int op;
		scanf("%d", &op);
		if (op == 1) {
			int l, r;
			ll c;
			scanf("%d %d %lld", &l, &r, &c);
			update(1, l, r, 1, n, c);
		} else {
			int l, r;
			scanf("%d %d", &l, &r);
			printf("%lld\n", getSum(1, l, r, 1, n));
		}
	}
	return 0;
}
2023/10/9 22:50
加载中...