线段树总结
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;
}