(区间更新lazy)POJ3468 A Simple Problem with Integers

最新推荐文章于 2022-12-09 12:45:57 发布

原创最新推荐文章于 2022-12-09 12:45:57 发布 · 148 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#线段树 #区间更新

数据结构 - 线段树专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用Lazy标记优化区段更新操作的线段树实现方法，并通过POJ3468题目的示例代码详细展示了如何进行区间更新及查询。此方法能够有效地减少不必要的节点更新，提高效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门：(区间更新lazy)POJ3468 A Simple Problem with Integers

模板题。

代码：

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=1e5+10;
int a[maxn];
typedef long long ll;

struct Tree{
    int l,r;
    ll sum,lazy;
    void update(int x){
        lazy+=x;
        sum+=1LL*(r-l+1)*x;
    }
}tree[maxn<<2];
int n,q;

void push_up(int x){
    tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum;
}

void push_down(int x){
    int lazyval=tree[x].lazy;
    if(lazyval){
        tree[x<<1].update(lazyval);
        tree[x<<1|1].update(lazyval);
        tree[x].lazy=0;
    }
}

void build(int x,int l,int r){
    tree[x].l=l,tree[x].r=r;
    tree[x].sum=tree[x].lazy=0;
    if(l==r){
        tree[x].sum=a[l];
        return ;
    }
    int mid=l+((r-l)>>1);
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    push_up(x);
}

void update(int x,int l,int r,int val){
    int L=tree[x].l,R=tree[x].r;
    if(l<=L&&R<=r){
        tree[x].update(val);
        return ;
    }
    push_down(x);
    int mid=L+((R-L)>>1);
    if(mid>=l) update(x<<1,l,r,val);
    if(mid<r) update(x<<1|1,l,r,val);
    push_up(x);
}

ll query(int x,int l,int r){
    int L=tree[x].l,R=tree[x].r;
    if(l<=L&&R<=r) return tree[x].sum;
    push_down(x);
    ll ans=0;
    int mid=L+((R-L)>>1);
    if(mid>=l) ans+=query(x<<1,l,r);
    if(mid<r) ans+=query(x<<1|1,l,r);
    push_up(x);
    return ans;
}


int main(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    build(1,1,n);
    char op;
    int a,b;
    ll c;
    while(q--){
        scanf(" %c",&op);
        if(op=='Q'){
            scanf("%d%d",&a,&b);
            printf("%lld\n",query(1,a,b));
        }else{
            scanf("%d%d%lld",&a,&b,&c);
            update(1,a,b,c);
        }
    }
    return 0;
}