给出一个n个元素的数组,设计一个数据结构,支持以下两种操作。
- Add(L,R,v):把
的值全部增加v。
- Query(L,R):计算子序列
的元素和、最小值和最大值。
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 10005;
const int inf = 0x3f3f3f3f;
int sumv[maxn<<2], minv[maxn<<2], maxv[maxn<<2], addv[maxn<<2];
// 维护结点o,它对应区间[L,R]
void maintain(int o, int l, int r) {
sumv[o] = minv[o] = maxv[o] = 0;
if (l < r) { // 考虑左右子树
int lc = o * 2, rc = o * 2 + 1;
sumv[o] = sumv[lc] + sumv[rc];
minv[o] = min(minv[lc], minv[rc]);
maxv[o] = max(maxv[lc], maxv[rc]);
}
// 考虑add操作
sumv[o] += addv[o] * (r - l + 1);
minv[o] += addv[o];
maxv[o] += addv[o];
}
int y1, y2;
int _sum, _min, _max; // 全局变量,目前位置的最小值、最大值和累加和
void query(int o, int l, int r, int add) {
if (y1 <= l && r <= y2) { // 递归边界:用边界区间的附加信息更新答案
_sum += sumv[o] + add * (r - l + 1);
_min = min(_min, minv[o] + add);
_max = max(_max, maxv[o] + add);
} else { // 递归统计,累加参数add
int m = (l + r) / 2;
int lc = o * 2, rc = o * 2 + 1;
if (y1 <= m) query(lc, l, m, add + addv[o]);
if (m < y2) query(rc, m + 1, r, add + addv[o]);
}
}
int v;
void update(int o, int l, int r) {
int lc = o * 2, rc = o * 2 + 1;
if (y1 <= l && r <= y2) { // 递归边界
addv[o] += v; // 累加边界的add值
} else {
int m = (l + r) / 2;
if (y1 <= m) update(lc, l, m);
if (m < y2) update(rc, m + 1, r);
}
maintain(o, l, r); // 递归结束前重新计算本结点的附加信息
}
int main() {
int n;
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d", &v);
y1 = y2 = i;
update(1, 1, n);
}
int op;
while (scanf("%d", &op) == 1) {
if (op) { // Query(L,R)
scanf("%d%d", &y1, &y2);
_sum = 0;
_min = inf;
_max = -inf;
query(1, 1, n, addv[1]);
printf("%d %d %d\n", _sum, _min, _max);
} else { // Add(L,R,v)
scanf("%d%d%d", &y1, &y2, &v);
update(1, 1, n);
}
}
return 0;
}