线段树模板 lazy tag 区间修改 区间和查询

#include<bits/stdc++.h>
using namespace std;
//区间修改 查询区间和 
const int N=2e5+10;
//结构体 可以用数组  ~~刚开始就学的数组~~
struct kl{
 int l,r;//左端点 右端点
 int sum,lazy;//区间和 标记
}a[N<<2]; 
int s[N];//接收数据的数组
//建树
void build(int node,int st,int en){
 a[node].l=st,a[node].r=en;//先赋左右端点
 if(st==en) {
 a[node].sum=s[st]; return ; //单点
 }
 int mid=(st+en)/2;
 //左右子树
 build(node*2,st,mid);
 build(node*2+1,mid+1,en);
 a[node].sum=a[node*2+1].sum+a[node*2].sum;
}
//将左右子树的值进行修改
void push(int node){
 if(a[node].lazy){
  a[node*2].sum+=a[node].lazy*(a[node*2].r-a[node*2].l+1);//左子树区间和
  a[node*2+1].sum+=a[node].lazy*(a[node*2+1].r-a[node*2+1].l+1);//右
  a[node*2].lazy+=a[node].lazy;//因为可能需要向下传递
  a[node*2+1].lazy+=a[node].lazy;
  a[node].lazy=0;//这个点实现完影响 归0
 }
}
//修改操作
void add(int node,int l,int r,int x){//
 if(l<=a[node].l&&r>=a[node].r){
 //根节点修改
  a[node].lazy+=x;
  a[node].sum+=x*(a[node].r-a[node].l+1);
  return ;//直接返回
 }
 push(node);//修改下一层子树
 int mid=(a[node].l+a[node].r)/2;
 if(l<=mid) add(node*2,l,r,x);//说明mid左边还有需要修改的结点
 if(r>mid) add(node*2+1,l,r,x);//同理 右
 a[node].sum=a[node*2].sum+a[node*2+1].sum;
}
//查询操作
int ask(int node,int l,int r){
 if(l<=a[node].l&&r>=a[node].r)  return a[node].sum;
 push(node); 
 int mid=(a[node].l+a[node].r)/2,res=0;
 if(l<=mid) res+=ask(node*2,l,r);
 if(r>mid)  res+=ask(node*2+1,l,r);
 return res;
} 
int main(){
 int n;
 cin>>n;
 for(int i=1;i<=n;i++) cin>>s[i];
 build(1,1,n);
 add(1,1,3,1);
 cout<<ask(1,1,3);
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值