A Simple Problem with Integers POJ - 3468 线段树区间修改+区间查询

题目链接:https://vjudge.net/problem/POJ-3468
题意:区间修改和查询。
思路:区间修改时一定要下传标记,这样才能更新上层节点的值,否则会出错。
如 5 3
1 1 1 1 1
C 1 2 1
C 2 3 1
Q 1 2
如果在区间修改时不下传标记,当对执行C 2 3 1时,此时4号节点也就是1和2的父节点的lazy没有下传导致pushup操作时4号节点的sum会出错。

#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
#include <iostream>
#include <cmath>
#include <set>
#include <fstream>
#include <list>
#include <stack>
#include <map>
using namespace std;
const long long maxn=1e5+5;
long long p[maxn];
struct NODE
{
    long long l,r;
    long long lazy,sum;
    NODE()
    {
        l=r=lazy=sum=0;
    }
} a[maxn*4];
void pushup(long long k)
{
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void build(long long l,long long r,long long k)
{
    a[k].l=l;
    a[k].r=r;
    if(l==r)
    {
        a[k].sum=p[l];
        return ;
    }
    long long mid=(l+r)/2;
    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
    pushup(k);
}
void pushdown(long long k)
{
    if(a[k].l==a[k].r)
    {
        a[k].lazy=0;
        return ;
    }
    a[k*2].sum+=(a[k*2].r-a[k*2].l+1)*a[k].lazy;
    a[k*2+1].sum+=(a[k*2+1].r-a[k*2+1].l+1)*a[k].lazy;
    a[k*2].lazy+=a[k].lazy;
    a[k*2+1].lazy+=a[k].lazy;
    a[k].lazy=0;
}
void add(long long l,long long r,long long L,long long R,long long k,long long val)
{
    if(l>R||r<L)
        return ;
    if(l>=L&&r<=R)
    {
        a[k].sum+=(a[k].r-a[k].l+1)*val;
        a[k].lazy+=val;
        return ;
    }
    long long mid=(l+r)/2;
    pushdown(k);
    add(l,mid,L,R,k*2,val);
    add(mid+1,r,L,R,k*2+1,val);
    pushup(k);
}
long long query(long long l,long long r,long long L,long long R,long long k)
{
    if(l>R||r<L)
        return 0;
    pushdown(k);
    if(l>=L&&r<=R)
        return a[k].sum;
    long long mid=(l+r)/2;
    long long cnt=0;
    cnt+=query(l,mid,L,R,k*2);
    cnt+=query(mid+1,r,L,R,k*2+1);
    return cnt;
}
int main()
{
    memset(a,0,sizeof(a));
    long long n,q;
    scanf("%lld%lld",&n,&q);
    for(long long i=1; i<=n; i++)
    {
        scanf("%lld",&p[i]);
    }
    build(1,n,1);
    char op[10];
    long long x,y;
    long long z;
    while(q--)
    {
        scanf("%s",op);
        if(op[0]=='Q')
        {
            scanf("%lld%lld",&x,&y);
            printf("%lld\n",query(1,n,x,y,1));
        }
        else
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            add(1,n,x,y,1,z);
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值