线段树模板

#include <bits/stdc++.h>
#define _for(i,a,b) for(int i=a;i<b;++i)
#define _rep(i,a,b) for(int i=a;i<=b;++i)
#define FREE freopen("in.txt","r",stdin)
#define FREO freopen("out.txt","w",stdout)
#define MAXN 100007
#define ls l, m, rt << 1
#define rs m + 1, r, rt << 1 | 1
using namespace std;
int Sum[MAXN << 2], Add[MAXN << 2];//Sum求和,Add为懒惰标记
int A[MAXN], n;
//PushUp函数更新节点信息,这里指求和
void PushUp(int rt) { Sum[rt] = Sum[rt << 1] + Sum[rt << 1 | 1]; }
//Build函数建树
void Build(int l, int r, int rt) {	//l,r表示当前结点区间,rt表示当前结点编号
	if (l == r) {//若到达叶节点
		Sum[rt] = A[l];
		return;
	}
	int m = (l + r) >> 1;
	//左右递归
	Build(l, m, rt << 1);
	Build(m + 1, r, rt << 1 | 1);
	PushUp(rt);
}
//点修改
void Update(int L, int C, int l, int r, int rt) {//l,r表示当前结点区间,rt表示当前结点编号
	if (l == r) {
		Sum[rt] += C;
		return;
	}
	int m = (l + r) >> 1;
	//根据条件判断往左子树调用还是右
	if (L <= m) Update(L, C, l, m, rt << 1);
	else Update(L, C, m + 1, r, rt << 1 | 1);
	PushUp(rt);
}
//区间修改
void PushDown(int rt, int ln, int rn);
void Update(int L, int R, int C, int l, int r, int rt) {//L,R表示操作区间,l,r表示当前结点区间,rt表示当前结点编号
	if (L <= l&&r <= R) {//如果本区间完全在操作区间[L,R]以内
		Sum[rt] += C*(r - l + 1);//更新数字和,向上保持正确
		Add[rt] += C;//增加Add标记,表示本区间的Sum正确,子区间的Sum仍需要根据Add的值来调整
		return;
	}
	int m = (l + r) >> 1;
	PushDown(rt, m - l + 1, r - m);//下推标记
	//这里判断左右子树跟[L,R]有无交集,有交集才递归
	if (L <= m) Update(L, R, C, l, m, rt << 1);
	if (R > m) Update(L, R, C, m + 1, r, rt << 1 | 1);
	PushUp(rt);//更新本节点信息
}
//区间查询
//1.下推标记的函数
void PushDown(int rt, int ln, int rn) {
	//ln,rn为左子树,右子树的数字数量
	if (Add[rt]) {
		//下推标记
		Add[rt << 1] += Add[rt];
		Add[rt << 1 | 1] += Add[rt];
		//修改子结点的Sum使之与对应的Add相对应
		Sum[rt << 1] += Add[rt] * ln;
		Sum[rt << 1 | 1] += Add[rt] * rn;
		//清除本节点标记
		Add[rt] = 0;
	}
}
//2.区间查询函数
int Query(int L, int R, int l, int r, int rt) {//L,R表示操作区间,l,r表示当前结点区间,rt表示当前结点编号
	if (L <= l&&r <= R) {
		//在区间内,直接返回
		return Sum[rt];
	}
	int m = (l + r) >> 1;
	PushDown(rt, m - l + 1, r - m);
	//累计答案
	int Ans = 0;
	if (L <= m) Ans += Query(L, R, l, m, rt << 1);
	if (R > m) Ans += Query(L, R, m + 1, r, rt << 1 | 1);
	return Ans;
}