0
1
mirror of https://git.sb/baoshuo/OI-codes.git synced 2024-12-13 06:26:26 +00:00
OI-codes/Luogu/P3372/P3372.cpp

120 lines
2.6 KiB
C++

#include <bits/stdc++.h>
using namespace std;
/**
* 线段树节点
*/
struct node {
int l, r;
long long s, d;
node() {
l = r = s = d = 0;
}
node(int _l, int _r) {
l = _l;
r = _r;
s = d = 0;
}
} tr[100005 << 2];
int n, m, op, x, y, k, a[100005];
/**
* 上传区间和
* @param u 父节点下标
*/
void pushup(int u) {
tr[u].s = tr[u << 1].s + tr[u << 1 | 1].s;
}
/**
* 下放懒标记
* @param u 父节点下标
*/
void pushdown(int u) {
if (!tr[u].d) return;
// 处理左子树
tr[u << 1].d += tr[u].d;
tr[u << 1].s += (tr[u << 1].r - tr[u << 1].l + 1) * tr[u].d;
// 处理右子树
tr[u << 1 | 1].d += tr[u].d;
tr[u << 1 | 1].s += (tr[u << 1 | 1].r - tr[u << 1 | 1].l + 1) * tr[u].d;
// 清除懒标记
tr[u].d = 0;
}
/**
* 建立线段树
* @param u 根节点下标
* @param l 左端点
* @param r 右端点
*/
void build(int u, int l, int r) {
tr[u] = node(l, r);
if (l == r) {
tr[u].s = a[l];
return;
}
int mid = l + r >> 1;
build(u << 1, l, mid);
build(u << 1 | 1, mid + 1, r);
pushup(u);
}
/**
* 修改区间
* @param u 父节点下标
* @param l 左端点
* @param r 右端点
* @param d 增加的值
*/
void modify(int u, int l, int r, int d) {
if (tr[u].l >= l && tr[u].r <= r) { // 被包含直接修改
tr[u].d += d;
tr[u].s += (tr[u].r - tr[u].l + 1) * d;
return;
}
int mid = tr[u].l + tr[u].r >> 1;
pushdown(u); // 下放懒标记
if (l <= mid) modify(u << 1, l, r, d); // 和左侧有交集
if (r > mid) modify(u << 1 | 1, l, r, d); // 和右侧有交集
pushup(u); // 上传新信息
}
/**
* 区间查询
* @param u 父节点
* @param l 左端点
* @param r 右端点
*/
long long query(int u, int l, int r) {
if (tr[u].l >= l && tr[u].r <= r) { // 被包含直接返回
return tr[u].s;
}
int mid = tr[u].l + tr[u].r >> 1;
long long s = 0;
pushdown(u); // 下放懒标记
if (l <= mid) s += query(u << 1, l, r); // 和左侧有交集
if (r > mid) s += query(u << 1 | 1, l, r); // 和右侧有交集
return s;
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
build(1, 1, n);
for (int i = 0; i < m; i++) {
cin >> op >> x >> y;
if (op == 1) {
cin >> k;
modify(1, x, y, k);
} else if (op == 2) {
cout << query(1, x, y) << endl;
}
}
return 0;
}