题意:包括两个操作:1、将[a.b]上的数字加上v;2、查询区间[a,b]上的和
下面的介绍是下解题思路:
首先介绍 lazy-tag思想:用一个变量记录每一个线段树节点的变化值,当这部分线段的一致性被破坏我们就将这个变化值传递给子区间,大大增加了线段树的效率。
比如现在需要对[a,b]区间值进行加c操作,那么就从根节点[1,n]开始调用update函数进行操作,如果刚好执行到一个子节点,它的节点标记为o,这时tree[o].l == a && tree[o].r == b 这时我们可以一步更新此时o节点的sum[o]的值,sumo] += c * (tree[o].r - tree[o].l + 1),注意关键的时刻来了,如果此时按照常规的线段树的update操作,这时候还应该更新o子节点的sum[]值,而Lazy思想恰恰是暂时不更新o子节点的sum[]值,到此就return,直到下次需要用到o子节点的值的时候才去更新,这样避免许多可能无用的操作,从而节省时间 。
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<stdio.h>
#include<stdlib.h>
#include<ctype.h>
#include<time.h>
#include<math.h>
#define ll __int64
#define N 100005
#define inf 0x7ffffff
#define eps 1e-9
#define pi acos(-1.0)
#define P system("pause")
using namespace std;
struct Node
{
int l,r;
ll sum,add;
}tree[N<<2];
ll a[N];
void PushUp(int o)//用于更新根节点
{
tree[o].sum = tree[2*o].sum + tree[2*o+1].sum;
}
void build(int o,int l,int r)
{
tree[o].l = l;
tree[o].r = r;
tree[o].add = 0;
if(l == r)
{
tree[o].sum = a[l];
return;
}
int m = (l+r)/2;
build(2*o,l,m);
build(2*o+1,m+1,r);
PushUp(o);
}
void PushDown(int o)//用于更新子节点
{
int m = (tree[o].r-tree[o].l+1);
if( tree[o].add)
{
tree[2*o].add += tree[o].add;
tree[2*o+1].add += tree[o].add;
tree[2*o].sum += tree[o].add*(m - (m>>1));
tree[2*o+1].sum += tree[o].add*(m>>1);
tree[o].add = 0;
}
}
void update(int o,int x,int y,int v)
{
if(x == tree[o].l && tree[o].r == y)
{
tree[o].add += v;//这个标记用来延迟更新
tree[o].sum += (__int64)v*(y-x+1);
return ;
}
if(tree[o].l == tree[o].r) return;
PushDown(o);
int m = (tree[o].r+tree[o].l)/2;
if(y <= m)
update(2*o,x,y,v);
else if(x > m)
update(2*o+1,x,y,v);
else {
update(2*o, x, m, v);
update(2*o+1, m+1, y, v);
}
PushUp(o);
}
ll query(int o,int x,int y)
{
if(tree[o].l == x && tree[o].r == y)
return tree[o].sum;
PushDown(o);
ll res = 0;
int m = (tree[o].l+tree[o].r)/2;
if(y <= m) res += query(2*o,x,y);
else if(x > m) res += query(2*o+1,x,y);
else {
res += query(2*o,x,m);
res += query(2*o+1,m+1,y);
}
return res;
}
int main()
{
//freopen("input.txt","r",stdin);
//freopen("output.txt","w",stdout);
int n,m;
while(scanf("%d%d",&n,&m) != EOF)
{
int i;
for(i = 1; i <= n; i++)
scanf("%I64d",&a[i]);
build(1,1,n);
char str[5];
while(m--)
{
scanf("%s",str);
int x,y,z;
if(str[0] == 'C')
{
scanf("%d%d%d",&x,&y,&z);
update(1,x,y,z);
}
if(str[0] == 'Q')
{
scanf("%d%d",&x,&y);
printf("%I64d\n",query(1,x,y));
}
}
}
return 0;
}
最后还有几点补充,
在update函数中,if(tree[o].l == x && y == tree[o].r) 这里就是用到Lazy思想的关键时刻 正如上面说提到的,这里首先更新该节点的sum[o]值,然后更新该节点具体每个数值应该加多少即add[o]的值,注意此时整个函数就运行完了,直接return,而不是还继续向子节点继续更新,这里就是Lazy思想,暂时不更新子节点的值。那么什么时候需要更新子节点的值呢?答案是在某部分update操作的时候需要用到那部分没有更新的节点的值的时候,这时就掉用PushDown()函数更新子节点的数值。
对于PushDown()函数,就是从当前根节点rt向下更新每个子节点的值,这段代码读者可以自己好好理解,这也是Lazy的关键。