线段树模板

@

目录

• 1、线段树实现方式

1、线段树实现方式

总之就是二分区间，分为左右儿子存
空间n*4，时间mlogn

模板：

#include<iostream>
#include<cstdio>
using namespace std;
const long long maxn=1e6+5;
long long a[maxn],d[maxn<<2],tag[maxn];
long long read(){
    long long x=0;char ch=getchar();bool pos=1;
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') pos=0;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';
    return pos?x:-x;
}
long long n,m; 
inline void push_up(long long now){
    d[now]=d[now<<1]+d[now<<1|1];
}
inline void push_down(long long nl,long long nr,long long now){
    long long mid=(nl+nr)>>1;
    tag[now<<1]+=tag[now];
    d[now<<1]+=tag[now]*(mid-nl+1);
    tag[now<<1|1]+=tag[now];
    d[now<<1|1]+=tag[now]*(nr-(mid+1)+1);
    tag[now]=0;
}
inline void build(long long l,long long r,long long now){
    if(l==r){
        d[now]=a[l];
        return;
    }
    long long mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
    push_up(now);
}
inline void change(long long l,long long r,long long nl,long long nr,long long now,long long w){
    if(nl>=l&&nr<=r){
        d[now]+=w*(nr-nl+1); //注意修改的东西
        tag[now]+=w;
        return;
    }
    push_down(nl,nr,now);
    long long mid=(nl+nr)>>1; //mid是nr和nr的中间，也就是这个储存区间的中间，但是要和查询区间来比较，检查是否在儿子中有查询区间的一部分
    if(l<=mid) change(l,r,nl,mid,now<<1,w);	
    if(r>mid) change(l,r,mid+1,nr,now<<1|1,w);
    push_up(now);
    return;
}
inline long long query(long long l,long long r,long long nl,long long nr,long long now){
    long long ans=0;
    if(nl>=l&&nr<=r) return d[now];
    push_down(nl,nr,now);
    long long mid=(nl+nr)>>1;	//基本同change
    if(l<=mid) ans+=query(l,r,nl,mid,now<<1);
    if(r>mid) ans+=query(l,r,mid+1,nr,now<<1|1);
    return ans;
}
int main(){
    n=read();m=read();
    for(long long i=1;i<=n;i++) a[i]=read();
    build(1,n,1);
    long long p;
    for(long long i=1;i<=m;i++){
        p=read();
        if(p==1){
            long long x=read(),y=read(),k=read();
            change(x,y,1,n,1,k);
        }else{
            long long x=read(),y=read();
            cout<<query(x,y,1,n,1)<<endl;
        }
    }
    return 0;
}