zoukankan      html  css  js  c++  java
  • luogu_3372 【模板】线段树 1

    #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
    ",query(1,k,l,r,1));
            }
        }
        return 0;
    }
    

      

  • 相关阅读:
    python中xrange和range的异同
    Python:使用threading模块实现多线程编程
    python Queue模块
    Python中pass语句的作用
    Python的作用域
    eclipse颜色配置
    protobuf
    python调试总结
    chardet安装
    Windows下搭建PHP开发环境
  • 原文地址:https://www.cnblogs.com/codetogether/p/7638481.html
Copyright © 2011-2022 走看看