线段树 最全注释(延迟标记)

 洛谷 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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值