萌新刚学OI1s(刚学分块),这题能用分块做吗
查看原帖
萌新刚学OI1s(刚学分块),这题能用分块做吗
253738
听取MLE声一片楼主2021/2/16 13:31

RT,70分,后三个点1.20sTLE,不过看好像有dalao用分块过了,求调。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<complex>
#define ll long long 
using namespace std;
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
const int N=100010;
int n,m,t,L[N],R[N],pos[N];
ll a[N],sum[N],add[N];
void uplate(int l,int r,ll 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<=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);
	}
}
ll query(int l,int r){
	int p=pos[l],q=pos[r];
	ll 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]*(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;
}
int main()
{
	n=read(),m=read();
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	t=sqrt(n);
	for(int i=1;i<=t;i++){
		L[i]=(i-1)*sqrt(t)+1;
		R[i]=i*sqrt(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];
		}
	while(m--){
		int x=read(),l=read(),r=read();
		ll d;
		if(x==1){
			scanf("%lld",&d);
			uplate(l,r,d);
		}
		else printf("%lld\n",query(l,r));
	}
	return 0;
}

2021/2/16 13:31
加载中...