分块做的,后面6个点全WA了,求助大佬QWQ
查看原帖
分块做的,后面6个点全WA了,求助大佬QWQ
174784
梦星之光楼主2020/12/19 15:44
#include<bits/stdc++.h>
using namespace std;
#define MAXN 500002
int n,m,st[MAXN],ed[MAXN],pos[MAXN],sum[2002],add[2002],num;
long long x,y,z,a[MAXN];
void change(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 ask(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()
{
	cin>>n>>m;
	int block=sqrt(n);
	int t=n/block;
	if(n%block) 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<=n;i++) cin>>a[i];
	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>>num;
		if(num==1)
		{
			cin>>x>>y>>z;
			change(x,y,z);
		}
		else if(num==2)
		{
			cin>>x;
			a[1]+=x;
			sum[pos[1]]+=x;
		}
		else if(num==3)
		{
			cin>>x;
			a[1]-=x;
			sum[pos[1]]-=x;
		}
		else if(num==4)
		{
			cin>>x>>y;
			cout<<ask(x,y)<<endl;
		}
		else cout<<a[1]<<endl;
	}
	return 0;
}
2020/12/19 15:44
加载中...