#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<vector>
#include<string.h>
#include<map>
#include<cmath>
#include<queue>
#include<stack>
#define ll long long
#define INF 0x7fffffff
#define MAX 0x3f3f3f3f
#define maxn 100005
#define ull unsigned long long
using namespace std;
int a[100005];
//树用结构体表示
struct tree{
int l,r;
ll pre,add;//pre 该节点维护的值,add 懒标记
}t[400050];
//建树
void bulid(int p,int l,int r){
t[p].l=l;//p节点维护的区间为l到r
t[p].r=r;
if(l==r){//特殊情况l=r 这个区间就只有一个数
t[p].pre=a[l];
return;
}
int mid=l+r>>1;
bulid(p*2,l,mid);//左子树
bulid(p*2+1,mid+1,r);//右子树
t[p].pre=t[p*2].pre+t[p*2+1].pre;//维护的值等于左儿子加右儿子
}
//懒标记
void dd(int p){
if(t[p].add){//如果懒标记不为0,就将其下传,修改左右儿子维护的值
t[p*2].pre+=t[p].add*(t[p*2].r-t[p*2].l+1);
t[p*2+1].pre+=t[p].add*(t[p*2+1].r-t[p*2+1].l+1);
t[p*2].add+=t[p].add;//为左右儿子加上懒标记
t[p*2+1].add+=t[p].add;
t[p].add=0;//既已下传,将该节点的懒标记清0
}
}
//区间修改
void change(int p,int x,int y,int z){
if(x<=t[p].l && y>=t[p].r){//要修改的区间覆盖了当前节点,就对其进行修改
t[p].pre+=(ll)z*(t[p].r-t[p].l+1);//注意加一
t[p].add+=z;//懒标记
return;
}
dd(p);//下传懒标记,继续向下找
int mid=t[p].l+t[p].r>>1;
if(x<=mid) change(p*2,x,y,z);//如果要修改的区间覆盖了左儿子,就修改左儿子
if(y>mid) change(p*2+1,x,y,z);//如果要修改的区间覆盖了右儿子,就修改右儿子
t[p].pre=t[p*2].pre+t[p*2+1].pre;//最终维护的值等于左儿子的值+右儿子的值
}
//区间查询
ll findd(int p,int x,int y){
if(x<=t[p].l&&y>=t[p].r)
return t[p].pre;//如果被覆盖,就返回维护的值
dd(p);//下传懒标记
int mid=t[p].l+t[p].r>>1;
ll ans=0;
if(x<=mid) ans+=findd(p*2,x,y);//返回左右儿子的和
if(y>mid) ans+=findd(p*2+1,x,y);
return ans;
}
int main(){
int n,m,i,q,x,y,z;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
bulid(1,1,n);//头节点编号是1,维护区间是1到n
for(i=1;i<=m;i++)
{
scanf("%d",&q);
if(q==1){
scanf("%d%d%d",&x,&y,&z);
change(1,x,y,z);//从头节点开始,要修改的区间,区间每个数加z
}
else{
scanf("%d%d",&x,&y);
cout<<findd(1,x,y)<<endl;//从头节点开始,查询区间各数之和
}
}
return 0;
}
线段树基本操作
最新推荐文章于 2022-10-24 22:56:43 发布