无lazy标记版
线段树lazy标记模板题
lazy标记怎么说呢,就是如果要修改的区间是[1,4]就在线段树中暂时标记被
区间[1,4]完全覆盖的节点而不去修改次节点以下的节点,如若[1,4]被标记,不会继续向下递归标记[1,4]的子节点[1,2],[3,4],使得时间复杂度减小到logn,但如果我们要查询【1,4】之后的节点怎么办呢,这个时候就要引入push-down操作。
代码:
void push_down(int p,int lf,int rt){//这是一个下传操作,更改子节点
if(lazy[p]){
int mid=(lf+rt)>>1;
lazy[p<<1]+=lazy[p];
lazy[p<<1|1]+=lazy[p];
ans[p<<1]+=lazy[p]*(mid-lf+1);
ans[p<<1|1]+=lazy[p]*(rt-(mid+1)+1);
lazy[p]=0;
}
}
如果当前区间被标记但还要继续递归下去,只需在下一次递归开始之前push-down,向下标记子节点即可。
区间修改代码:
void update(int p,int lf,int rt,int l,int r,int val){//单点修改
if(lf>=l&&rt<=r){//当前查询区间被目标修改区间完全覆盖则标记
lazy[p]+=val;
ans[p]+=(rt-lf+1)*val; return ;
}
push_down(p,lf,rt);//判断当前查询区间有无lazy标记
int mid=(lf+rt)>>1;
if(l<=mid) update(lson,l,r,val);//若当前区间的左区间存在被覆盖的情况则搜左区间
if(r>mid) update(rson,l,r,val);//若当前区间的右区间存在被覆盖的情况则搜右区间
push_up(p);//合并更新后节点值
}
区间求和时因为不知道线段树是否仍然存在被标记的点所以在每次递归之前仍需进行push-down操作以防万一,其他的与无lazy标记时求和一样。
区间求和代码:
int sum(int p,int lf,int rt,int l,int r){//区间求和
if(lf>=l&&rt<=r){//当前查询区间被目标区间完全覆盖
return ans[p];
}
int he=0;//判断当前查询区间有无lazy标记
push_down(p,lf,rt);
int mid=(lf+rt)>>1;
if(l<=mid) he+=sum(lson,l,r);//若当前区间的左区间存在被覆盖的情况则搜左区间
if(r>mid) he+=sum(rson,l,r);//若当前区间的右区间存在被覆盖的情况则搜右区间
return he;
}
完整代码:
#include <iostream>
using namespace std;
#define int long long
#define lll 1000001
#define lson p<<1,lf,mid
#define rson p<<1|1,mid+1,rt
int ans[lll<<2],A[lll],lazy[lll<<2]; int n,m;
void push_up(int p){
ans[p]=ans[p<<1]+ans[p<<1|1];
}
void push_down(int p,int lf,int rt){//这是一个下传操作,更改子节点
if(lazy[p]){
int mid=(lf+rt)>>1;
lazy[p<<1]+=lazy[p];
lazy[p<<1|1]+=lazy[p];
ans[p<<1]+=lazy[p]*(mid-lf+1);
ans[p<<1|1]+=lazy[p]*(rt-(mid+1)+1);
lazy[p]=0;
}
}
void build(int p,int lf,int rt){//建树
lazy[p]=ans[p]=0;
if(lf==rt) {
ans[p]=A[lf]; return ;
}
int mid=(lf+rt)>>1;
build(lson);
build(rson);
push_up(p);
}
void update(int p,int lf,int rt,int l,int r,int val){//单点修改
if(lf>=l&&rt<=r){//当前查询区间被目标修改区间完全覆盖则标记
lazy[p]+=val;
ans[p]+=(rt-lf+1)*val; return ;
}
push_down(p,lf,rt);//判断当前查询区间有无lazy标记
int mid=(lf+rt)>>1;
if(l<=mid) update(lson,l,r,val);//若当前区间的左区间存在被覆盖的情况则搜左区间
if(r>mid) update(rson,l,r,val);//若当前区间的右区间存在被覆盖的情况则搜右区间
push_up(p);//合并更新后节点值
}
int sum(int p,int lf,int rt,int l,int r){//区间求和
if(lf>=l&&rt<=r){//当前查询区间被目标区间完全覆盖
return ans[p];
}
int he=0;//判断当前查询区间有无lazy标记
push_down(p,lf,rt);
int mid=(lf+rt)>>1;
if(l<=mid) he+=sum(lson,l,r);//若当前区间的左区间存在被覆盖的情况则搜左区间
if(r>mid) he+=sum(rson,l,r);//若当前区间的右区间存在被覆盖的情况则搜右区间
return he;
}
void chaxun(int a){
if(a==1){
int o,p,q; cin>>o>>p>>q;
update(1,1,n,o,p,q);
}
if(a==2){
int x,y; cin>>x>>y;
int t=sum(1,1,n,x,y);
cout<<t<<endl;
}
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>A[i];
build(1,1,n);
for(int i=0;i<m;i++){
int a; cin>>a;
chaxun(a);
}
}