线段树模板
luogu P3372
#include <iostream>
using namespace std;
const int maxn = 100010;
int a[maxn+2],n,m;
struct tree
{
int l,r;
long long add,pre;
}t[maxn*4+2];
void build(int p,int l,int r)//建树
{
t[p].l=l,t[p].r=r;
if(l==r)
{
t[p].pre=a[l];
return ;
}
int mid = (l+r)>>1;
build(p<<1,l,mid);
build(p<<1|1,mid+1,r);
t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
void lazy_spread(int p)//懒标记
{
if(t[p].add)
{
t[p<<1].pre+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
t[p<<1|1].pre+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1);
t[p<<1].add+=t[p].add;
t[p<<1|1].add+=t[p].add;
t[p].add=0;
}
}
void change(int p,int x,int y,int z)//区间修改
{
if(t[p].l>=x&&t[p].r<=y)
{
t[p].pre+=(long long )z*(t[p].r-t[p].l+1);
t[p].add+=z;
return ;
}
lazy_spread(p);
int mid = (t[p].l+t[p].r)>>1;
if(x<=mid)change(p<<1,x,y,z);
if(y>mid)change(p<<1|1,x,y,z);
t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
long long ask(int p,int x,int y)//区间查询
{
if(t[p].l>=x&&t[p].r<=y)return t[p].pre;
lazy_spread(p);
int mid = (t[p].l+t[p].r)>>1;
long long ans = 0;
if(x<=mid)ans+=ask(p<<1,x,y);
if(y>mid)ans+=ask(p<<1|1,x,y);
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>a[i];
build(1,1,n);
for(int i=1;i<=m;i++)
{
int x,y,z;
cin>>z;
if(z-1)
{
cin>>x>>y;
cout<<ask(1,x,y)<<endl;
}
else
{
cin>>x>>y>>z;
change(1,x,y,z);
}
}
return 0;
}