#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);
}
线段树模板 lazy tag 区间修改 区间和查询
最新推荐文章于 2024-08-28 14:35:43 发布