题目传送门
给你n个数,m次操作,操作有两种,第一种,对于区间内每个数加上k,第二种输出区间内所有数的和。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
const int mod=1e9+7;
const int INF=0x7fffffff;
const ll LLINF=0x7fffffffffffffff;
const double EPS=1e-10;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define pii pair<int,int>
#define pll pair<ll,ll>
#define debug cout<<"debug"<<endl;
//#define int long long
int a[N];
struct tree
{
int l,r;
ll pre,add;
}t[N<<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 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(x<=t[p].l&&t[p].r<=y)
{
t[p].pre+=(ll)z*(t[p].r-t[p].l+1);
t[p].add+=z;
return ;
}
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;
}
ll ask(int p,int x,int y)
{
if(x<=t[p].l&&y>=t[p].r)
return t[p].pre;
spread(p);
int mid=(t[p].l+t[p].r)>>1;
ll ans=0;
if(x<=mid)
ans+=ask(p<<1,x,y);
if(y>mid)
ans+=ask(p<<1|1,x,y);
return ans;
}
signed main()
{
IOS;
//freopen("","r",stdin);
//freopen("","w",stdout);
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
build(1,1,n);
while(m--)
{
int q,x,y,z;
cin>>q;
if(q==1)
{
cin>>x>>y>>z;
change(1,x,y,z);
}
else
{
cin>>x>>y;
cout<<ask(1,x,y)<<endl;
}
}
}