#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;
}
}
}
标签:pr,线段,mid,addtag,tag,模板,ll,pl
From: https://www.cnblogs.com/zzzsacmblog/p/18117193