114514xxx @ 2024-08-13 15:53:36
#include<bits/stdc++.h>
#define int long long
const int N=5e6+25;
int a[N],sum[N],add[N];
int L[N],R[N];
int pos[N],n,m,k,x,y,opt;
using namespace std;
void add1(int l,int r,int d){
int p=pos[l],q=pos[r];
if(p==q){
for(int i=l;i<=r;i++)
a[i]+d;
sum[p]+=(r-l+1)*d;
}else{
for(int i=p+1;i<=q-1;i++)add[i]+=d;
for(int i=l;i<=R[p];i++)a[i]+=d;
sum[p]+=d*(R[p]-l+1);
for(int i=L[q];i<=r;i++)a[i]+=d;
sum[q]+=d*(r-L[q]+1);
}
}
int query(int l,int r){
int ans=0;
int p=pos[l],q=pos[r];
if(p==q){
for(int i=l;i<=r;i++)
ans+=a[i];
ans+=add[p]*(r-l+1);
}else{
for(int i=p+1;i<=q-1;i++)
ans+=sum[i]+add[i]*(R[i]-L[i]+1);
for(int i=l;i<=R[p];i++)ans+=a[i];
ans+=add[p]*(R[p]-l+1);
for(int i=L[q];i<=r;i++)ans+=a[i];
ans+=add[q]*(r-L[q]+1);
}
return ans;
}
signed main(){
std::ios::sync_with_stdio(0);
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>a[i];
int t=sqrt(n);
for(int i=1;i<=t;i++){
L[i]=(i-1)*t+1;
R[i]=i*t;
}
if(R[t]<n)t++,L[t]=R[t-1]+1,R[t]=n;
for(int i=1;i<=t;i++){
for(int j=L[i];j<=R[i];j++){
pos[j]=i;
sum[i]+=a[j];
}
}
for(int i=1;i<=m;i++){
cin>>opt>>x>>y;
if(opt==1){
cin>>k;
add1(x,y,k);
}else
cout<<query(x,y)<<endl;
}
return 0;
}
by 114514xxx @ 2024-08-13 16:23:30
已AC,此贴完结。