BZOJ3462 DZY Loves Math II

题目

首先如果(S)中含平方因子或者(sum p>x)，直接输出(0)

之后我们令(x)减去(sum p)，之后就变成了一个完全背包问题

背包的体积那么大可咋搞啊

我们注意到对于一个质数(p_i)，其能表示的数是(aS+bp_i(bp_i<S))的形式，两种方案不同当且仅当存在一个质数(a)不同或(b)不同，于是我们可以对于两部分分别计算乘法原理合并就好了

由于质因子个数不会超过(7)，所以所有质数(bp_i)的总和不会超过(7 imes S)，我们对这部分跑一个多重背包；对于一个询问(x)，我们枚举一下这一部分取得总和是多少，剩下的肯定是(S)的整数倍，用组合数分配一下就好了

代码

#include<bits/stdc++.h>
#define re register
#define LL long long
const int mod=1e9+7;
int m,S,Q,Sum,Sz,ans;LL n;
int pre[14000005],dp[14000005];
int f[200005],p[100005],Pr[105],inv[105];
inline int dqm(int x) {return x<0?x+mod:x;}
inline int qm(int x) {return x>=mod?x-mod:x;}
inline int C(LL n,int m) {
	if(m>n) return 0;
	int nw=inv[m];
	for(re LL i=n;m;--i,--m) nw=1ll*nw*(i%mod)%mod;
	return nw;
} 
int main() {
	scanf("%d%d",&S,&Q);int R=sqrt(S)+1;m=S;
	for(re int i=2;i<=R;i++) {
		if(!f[i]) p[++p[0]]=i;
		for(re int j=1;j<=p[0]&&p[j]*i<=R;++j) {
			f[p[j]*i]=1;if(i%p[j]==0) break;
		}
	}
	for(re int i=1;i<=p[0];i++) {
		if(S==1) break;if(S%p[i]) continue;
		S/=p[i];if(S%p[i]==0) {while(Q--)puts("0");return 0;}
		Pr[++Pr[0]]=p[i];Sum+=p[i];
	}
	if(S!=1) Pr[++Pr[0]]=S,Sum+=S;dp[0]=1;
	for(re int i=1;i<=Pr[0];++i) {
		Sz+=m-Pr[i];
		for(re int j=0;j<=Sz;++j) {
			if(j>=Pr[i]) pre[j]=qm(pre[j-Pr[i]]+dp[j]);else pre[j]=dp[j];
			dp[j]=dqm(pre[j]-(j<m?0:pre[j-m]));
		}
	}
	inv[0]=inv[1]=1;for(re int i=2;i<=Pr[0];++i) inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	for(re int i=2;i<=Pr[0];++i) inv[i]=1ll*inv[i]*inv[i-1]%mod;
	for(;Q;--Q,ans=0) {
		scanf("%lld",&n);if(n<Sum) {puts("0");continue;}n-=Sum;
		for(re int i=n%m,tot=0;i<=n&&i<=Sz;i+=m,++tot)
			ans=qm(ans+1ll*dp[i]*C(n/m-tot-1+Pr[0],Pr[0]-1)%mod);
		printf("%d
",ans);
	}
	return 0;
}