CF103D Time to Raid Cowavans(分块)

Problem

Solution

• 离线做，先按k的大小排序。
• 分块，对于k>sqrtn的部分暴力求和，对于每个不同的小于sqrtn的k，维护k个后缀和，对于这部分询问，二分或倍增的求出其在和数列的对应位置。
• 空间有限，vector存后缀和。

Code

#include<iostream>
#include<cstdio>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=3e5+5;
int n,m,a[maxn],sqrn;
long long ac[maxn];
bool inq[maxn];
vector<long long>num[2005];
struct query{
	int t,k,i;
}q[maxn];
bool cmp(const query &x,const query &y)
{
	if(x.k==y.k)
	return x.t<y.t;
	return x.k>y.k;
}
long long solve1(int t,int k)
{
	long long ans=0;
	for(int i=t;i<=n;i+=k)
	ans+=a[i];
	return ans;
}
long long solve2(int t,int k)
{
	if(!inq[k])
	{
		inq[k]=1;
		for(int i=0;i<=sqrn;i++)
		num[i].clear();
		for(int i=0;i<=k-1;i++)
		{
			long long ans=0;
			for(int j=n-i;j>0;j-=k)
			{
				ans+=a[j];
				num[i].push_back(ans);
			}
		}
	}
	int pos=(n-t)/k,logg=0,pos0,nw=0;
	pos0=(t+k*pos);pos=n-pos0;
	while(pos0-(1<<logg)*k>=t)
	logg++;
	logg--;
	for(int i=logg;i>=0;i--)
	{
		if(pos0-(1<<i)*k>=t)
		{
			pos0-=(1<<i)*k;
			nw+=(1<<i);
		}
	}
	return num[pos][nw];
}
int main()
{
	cin>>n;
	sqrn=sqrt(n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&q[i].t,&q[i].k);
		q[i].i=i;
	}
	sort(q+1,q+1+m,cmp);
	for(int i=1;i<=m;i++)
	{
		if(q[i].k>=sqrn)
		ac[q[i].i]=solve1(q[i].t,q[i].k);
		else
		ac[q[i].i]=solve2(q[i].t,q[i].k);
	}
	for(int i=1;i<=m;i++)
	cout<<ac[i]<<endl;
    return 0;	
}