bzoj3462: DZY Loves Math II

第一眼看题没看懂。。后来发现pi可以等于pi+1。。。

题目可化为给定S=p1+p2+...+pk，求p1*x1+p2*x2+..+pk*xk=n的方案数(x1,x2,x3,...>0)。

因为S<=2000000，所以k<=7。

不妨把xi分为ai=xi/(S/pi),bi=xi%(S/pi)。枚举m，p1*b1+p2*b2+...+pk*bk=n-m*S。这个可以用背包的方式预处理，剩下的可用插板法得到，时间复杂度O(sqrt(S)+S*k^2+q*k^2)。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define ll long long
#define P 1000000007
using namespace std;
int S,q,a[10],cnt,once,s;
ll f[15000005],last[15000005],sum[15000005],inv[10],n,Ans,rest,part;
ll Mod(ll a)
{
	if (a<0) return a+P;
	if (a>=P) return a-P;
	return a;
}
ll Get(ll x,ll y)
{
	x=x+y-1;y=y-1;x%=P;
	ll ans=1;
	for (ll i=1;i<=y;i++)
		ans=ans*(x-i+1)%P*inv[i]%P;
	return ans;
}
int main()
{
	scanf("%d%d",&S,&q);s=S;
	for (int i=2;i*i<=s;i++)
		if (s%i==0)
		{
			a[++cnt]=i,s/=i;
			if (s%i==0){while(q--)puts("0");return 0;}
		}
	if (s>1) a[++cnt]=s;
	for (int i=1;i<=cnt;i++)
		once+=a[i];
	f[0]=1;
	for (int i=1;i<=cnt;i++)
	{
		memcpy(last,f,sizeof(ll)*(S*(i-1)+1));
		for (int j=0;j<a[i];j++) sum[j]=f[j]=last[j];
		for (int j=a[i];j<=S*i;j++)
		{
			int k=j%a[i];
			if (j>=S) sum[k]=Mod(sum[k]-last[j-S]);
			sum[k]=Mod(sum[k]+last[j]);
			f[j]=sum[k];
		}
	}
	inv[1]=1;
	for (int i=2;i<=cnt;i++)
		inv[i]=(P-P/i)*inv[P%i]%P;
	while (q--)
	{
		scanf("%lld",&n);n-=once;
		Ans=0,rest=n%S,part=n/S;
		for (int i=0;i<cnt;i++)
		{
			if (n<i*S) break;
			Ans=(Ans+f[i*S+rest]*Get(part-i,cnt))%P;
		}
		printf("%lld\n",Mod(Ans));
	}
	return 0;
}