线段树模板

#include<iostream>
#include<map>
#include<algorithm>
#include<math.h>

using namespace std;
/*3372*/
typedef long long ll;
const int N = 1e5 + 10;
ll a[N] = { 0 };
ll tree[N << 2] = { 0 };
ll tag[N << 2] = { 0 };
ll ls(ll p) { return p << 1; }
ll rs(ll p) { return p << 1 | 1; }
void push_up(ll p)//更新节点，这里的p指的是线段树静态数组中的索引，也就是树的树叶
{
	tree[p] = tree[ls(p)] + tree[rs(p)];
}
void build(ll p, ll pl, ll pr)
//给出的是建树，需要p，p的左节点pl，p的右节点pr，一般调用是直接build(1,1,n),从第一个节点开始
{
    tag[p] = 0;//lazytag标记修改
	if (pl == pr) { tree[p] = a[pl]; return; }
	ll mid = (pl + pr) >> 1;
	build(ls(p), pl, mid);//左子树建立
	build(rs(p), mid + 1, pr);//右子树建立
	push_up(p);//p节点更新
}
void addtag(ll p, ll pl, ll pr, ll d)//更新：打上标签，作用：在某个区间被完全覆盖的时候可以使用addtag
{
	tag[p] += d;
	tree[p] += d * (pr - pl + 1);
}
void push_down(ll p, ll pl, ll pr)//不能覆盖的时候，把“addtag”中生成的标签tag传递给左右子树
{
	if (tag[p])//如果tag[p]不为0，就是被addtag打上标记的时候
	{
		ll mid = (pl + pr) >> 1;//一样，利用addtag传递
		addtag(ls(p), pl, mid, tag[p]);
		addtag(rs(p), mid + 1, pr, tag[p]);
		tag[p] = 0;
	}
}
void update(ll L, ll R, ll p, ll pl, ll pr, ll d)
{
	//L,R里的每个元素＋d
	if (L <= pl and pr <= R) { addtag(p, pl, pr, d); return; }//当完全覆盖，直接返回并打上标记
	push_down(p, pl, pr);//如果没有return掉，说明就需要分隔
	ll mid = (pl + pr) >> 1;
	if (L <= mid)update(L, R, ls(p), pl, mid, d);//左子树
	if (R > mid)update(L, R, rs(p), mid + 1, pr, d);//右子树
	push_up(p);
}
ll query(ll L, ll R, ll p, ll pl, ll pr)
{
	//查询区间L,R，p是当前节点（线段）的编号，pl，pr是节点p的线段区间
	if (pl >= L and pr <= R)return tree[p];//完全覆盖，以顶点代区间
	push_down(p, pl, pr);//不能就先用push_down处理tag
	ll res = 0;
	ll mid = (pl + pr) >> 1;
	if (L <=mid)res += query(L, R, ls(p), pl, mid);
	if (R > mid)res += query(L, R, rs(p), mid + 1, pr);
	return res;
}
int main()
{
	ll n, m; cin >> n >> m;
	for (ll i = 1; i <= n; i++)cin >> a[i];
	build(1, 1, n);
	while (m--)
	{
		ll q, L, R, d;
		cin >> q;
		if (q == 1)
		{
			cin >> L >> R >> d;
			update(L, R, 1, 1, n, d);
		}
		else
		{
			cin >> L >> R;
			cout << query(L, R, 1, 1, n)<<endl;
		}
	}
}