分块10分
查看原帖
分块10分
1009027
stils楼主2024/1/1 11:22
#include<bits/stdc++.h>
using namespace std;
const int N=100005;
int n,m,op;
int a[N],pos[N];
int sum[N],st[N],ed[N],add[N];
int t;
int block;
void update(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]+=d*(R-L+1);
	}
	else{
		for(int i=p+1;i<=q-1;i++){
			add[i]+=d;
			for(int i=L;i<=ed[p];i++)a[i]+=d;
			sum[p]+=d*(ed[p]-L+1);
			for(int i=st[q];i<=R;i++)a[i]+=d;
			sum[q]+=d*(R-st[q]+1);
		}
	}
}
long long query(int L,int R){
	int p=pos[L],q=pos[R];
	long long ans=0;
	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]*(ed[i]-st[i]+1);
		}
		for(int i=L;i<=ed[p];i++)ans+=a[i];
		ans+=add[p]*(ed[p]-L+1);
		for(int i=st[q];i<=R;i++)ans+=a[i];
		ans+=add[q]*(R-st[q]+1);
	}
	return ans;
}


int main(){
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	block=sqrt(n);
	t=n/block;
	if(n%block!=0)t++;
	for(int i=1;i<=t;i++){
		st[i]=(i-1)*block+1;
		ed[i]=i*block;
	}
	ed[t]=n;
	for(int i=1;i<=n;i++){
		pos[i]=(i-1)/block+1;
	}
	for(int i=1;i<=t;i++){
		for(int j=st[i];j<=ed[i];j++){
			sum[i]+=a[j];
		}
	}
	for(int i=1;i<=m;i++){
		cin>>op;
		if(op==1){
			int x,y,k;
			cin>>x>>y>>k;
			update(x,y,k);
		}
		if(op==2){
			int x,y;
			cin>>x>>y;
			cout<<query(x,y)<<endl; 
		}
	}
	
	return 0;
} 
2024/1/1 11:22
加载中...