poj3468(线段树成段更新模板题)

题意:包括两个操作: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的关键。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值