luogu_3372 【模板】线段树 1

最新推荐文章于 2024-12-26 15:04:08 发布

weixin_30344795

最新推荐文章于 2024-12-26 15:04:08 发布

阅读量62

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/codetogether/p/7638481.html

本文介绍了一种使用线段树进行区间更新与查询的方法，包括如何在更新操作中保持懒惰标记，并通过示例代码展示了具体实现过程。

#include <cstdio>
#include <iostream>
using namespace std;
const int N=1000010;
int n,m;
long long a[N],addv[N];

void pushdown(int o,int len){
    if(addv[o]){
        addv[o*2]+=addv[o];
        addv[o*2+1]+=addv[o];
        a[o*2]+=addv[o]*(len-len/2);
        a[o*2+1]+=addv[o]*(len/2);
        addv[o]=0;
    }
    return;
}

void pushup(int o){
    a[o]=a[o*2]+a[o*2+1];
    return;
}

void update(int l,int r,int L,int R,long long v,int o){
    if(L<=l && r<=R){addv[o]+=v; a[o]+=v*(r-l+1); return;}
    pushdown(o,r-l+1);
    int mid=(l+r)>>1;
    if(mid>=L)update(l,mid,L,R,v,o*2);
    if(mid<R)update(mid+1,r,L,R,v,o*2+1);
    pushup(o);
}

long long query(int l,int r,int L,int R,int o){
    long long ans=0;
    if(L<=l && r<=R)return a[o];
    pushdown(o,r-l+1);
    int mid=(l+r)>>1;
    if(mid>=L)ans+=query(l,mid,L,R,o*2);
    if(mid<R)ans+=query(mid+1,r,L,R,o*2+1);
    return ans;
}

int main(){
    int k=0;
    scanf("%d%d",&n,&m);
    while((1<<k)<n)k++;
    k=1<<k;
    for(int i=k;i<k+n;i++)scanf("%lld",&a[i]);
    for(int i=k-1;i;i--)a[i]=a[i*2]+a[i*2+1];
    while(m--){
        int x;
        scanf("%d",&x);
        if(x==1){
            int l,r; long long v;
            scanf("%d%d%lld",&l,&r,&v);
            update(1,k,l,r,v,1);
        }
        else {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%lld\n",query(1,k,l,r,1));
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/codetogether/p/7638481.html