洛谷 P3372 【模板】线段树 1
#include <bits/stdc++.h>
#define l(x) a[x].Left
#define r(x) a[x].Right
#define sum(x) a[x].Sum
#define add(x) a[x].add
#define MAXN 100010
using namespace std;
struct Tree {
int Left;//左区间
int Right;//右区间
long long add, Sum;//标记 总和
};
int n, m;
Tree a[4 * MAXN];
int waiting[MAXN];
void biuld(int p, int l, int r) {//p 树状数组序号 l 左区间 r 右区间
l(p) = l, r(p) = r;
if (l == r) {
sum(p) = waiting[l];// 给子叶赋值waiting
return;//到达子叶,返回
}
int mid = (l + r) / 2;
biuld(2 * p, l, mid);
biuld(2 * p + 1, mid + 1, r);
sum(p) = sum(2 * p) + sum(2 * p + 1);//求区间和
}//建线段树
void spread(int p) {
if (add(p)) { //节点p有标记(前提该子节点被修改同时打上了延迟标记)
sum(p * 2) += add(p) * (r(p * 2) - l(p * 2) + 1); //更新左子节点信息
//(r(p * 2) - l(p * 2) + 1)代表总个数 下面的一样
sum(p * 2 + 1) += add(p) * (r(p * 2 + 1) - l(p * 2 + 1) + 1); //更新右子节点信息
add(p * 2) += add(p); //给左子节点打延迟标记 可能会有原来的所以是加不是直接等于
add(p * 2 + 1) += add(p); //给右子节点打延迟标记
add(p) = 0;//清除 p 的标记,防止下一次在这里再次开始打标记
}
}//延迟标记函数
long long ask (int p, int l, int r) {//输出区间[l,r]的区间和
if (l <= l(p) && r >= r(p)) {
return sum(p);//为计数器献上Sum值!
}
spread(p); //下传延迟标记
int mid = (l(p) + r(p)) / 2;
long long ans = 0;
if (l <= mid) {
ans += ask(p * 2, l, r);
}
if (r > mid) {
ans += ask(p * 2 + 1, l, r);
}
return ans;
}//求区间和
void change (int p, int l, int r, int d) { //p 线段树下标 l 需要修改的左区间 r 需要修改的右区间 d 需要加的数
if (l <= l(p) && r >= r(p)) {//如果区间完全覆盖
sum(p) += (long long)d * (r(p) - l(p) + 1); //更新节点区间和
add(p) += d; //给节点打延迟标记
return;
}
spread(p);//如果并没有全覆盖,那么就下传延迟标记,直到被全覆盖
int mid = (l(p) + r(p)) / 2;
if (l <= mid) {
change(p * 2, l, r, d);
}
if (r > mid) {
change(p * 2 + 1, l, r, d);
}
sum(p) = sum(p * 2) + sum(p * 2 + 1);
}//修改区间和
int main() {
ios::sync_with_stdio(0);//大数据优化cin,cout
cin.tie(0);
cout.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; i++) {
cin >> waiting[i];
}//输完一块建
biuld(1, 1, n);
while (m--) {
int op, x, y;//x,y 都是序列
cin >> op;
if (op == 1) {
int k;
cin >> x >> y >> k;
change(1, x, y, k);
} else {
cin >> x >> y;
cout << ask(1, x, y) << endl;
}
}
return 0;
}