线段树,延迟标记
本题要点:
1、延迟标记:
当前的修改或者查询的区间范围是[l, r], 如果某个节点的区间 p 被 [l, r] 所覆盖(l <= tree[p].l && tree[p].r <= r),
并且这个节点p ,在后续再也没有访问到,这个节点的子节点的相关信息可以不修改。 如果 p节点后续访问到了,只有在访问到
的时候,先更新 p 节点的所有子节点信息。 至于延迟标记,可以在 节点的结构体中,增加 add 标记。
查询到某节点 p, 根据 add 标记, 先 spread§, 再做后续操作。
2、 本题,区间 [l, r] 每个数加上 v, 并且查询任意区间 [l, r] 所有元素的和。 因此,在 线段树节点结构体维护的信息
区间范围 l, r; 区间总和 sum; 区间延迟标记 add .
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MaxN = 5e5 + 10;
int a[MaxN];
int n, m;
struct segTree
{
int l, r;
long long sum, add;
}tree[MaxN * 4];
void build(int p, int l, int r)
{
tree[p].l = l, tree[p].r = r;
if(l == r)
{
tree[p].sum = a[l];
return;
}
int mid = (l + r) / 2;
build(p * 2, l, mid);
build(p * 2 + 1, mid + 1, r);
tree[p].sum = tree[p * 2].sum + tree[p * 2 + 1].sum;
}
void spread(int p) // 从节点p向下递归
{
if(tree[p].add)
{
tree[p * 2].sum += tree[p].add * (tree[p * 2].r - tree[p * 2].l + 1);
tree[p * 2 + 1].sum += tree[p].add * (tree[p * 2 + 1].r - tree[p * 2 + 1].l + 1);
tree[p * 2].add += tree[p].add; //给左孩子打上延迟标记
tree[p * 2 + 1].add += tree[p].add; //给右孩子打上延迟标记
tree[p].add = 0; //清除p的标记
}
}
void change(int p, int l, int r, int d)
{
if(l <= tree[p].l && r >= tree[p].r)
{
tree[p].sum += (long long)d * (tree[p].r - tree[p].l + 1); //更新节点信息
tree[p].add += d; //给节点打上延迟标记
return;
}
spread(p);
int mid = (tree[p].l + tree[p].r) / 2;
if(l <= mid)
{
change(p * 2, l, r, d);
}
if(r > mid)
{
change(p * 2 + 1, l, r, d);
}
tree[p].sum = tree[2 * p].sum + tree[2 * p + 1].sum;
}
long long ask(int p, int l, int r)
{
if(l <= tree[p].l && r >= tree[p].r)
{
return tree[p].sum;
}
spread(p);
int mid = (tree[p].l + tree[p].r) / 2;
long long val = 0;
if(l <= mid)
{
val += ask(p * 2, l, r);
}
if(r > mid)
{
val += ask(p * 2 + 1, l, r);
}
return val;
}
int main()
{
int cmd, x, y, v;
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
}
build(1, 1, n);
for(int i = 0; i < m; ++i)
{
scanf("%d", &cmd);
if(1 == cmd)
{
scanf("%d%d%d", &x, &y, &v);
change(1, x, y, v);
}else{
scanf("%d%d", &x, &y);
printf("%lld\n", ask(1, x, y));
}
}
return 0;
}
/*
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
*/
/*
11
8
20
*/