BZOJ3462: DZY Loves Math II

首先我们要对着题目多看几遍，会发现，题目要求的质数序列里面的数一定都是$S$的质因数！
然后再想想，会发现如果$S$有平方因子时一定无解（这些质数的$lcm$不可能为$S$）
那么问题就转化成了给一些质数，每种可以用若干次（至少一次），求和为n的有几种
把n减去每个质因子就能去掉至少一次的限制
考虑质因子个数，用计算器算一下易知是≤7的
易知这道题是要DP的，但是n很大，$10^{18}$级别，直接DP就挂了
因为$S$不大，考虑围绕$S$DP。
设最后序列是这样的：
$p1*c1+p2*c2+p3*c3....=n$
（看题解）易知这是一个经典模型
因为每个$p$都是$S$的因数，所以$pi*ci=KS+kpi$
可以将$ci$拆分成$ai*(S/pi)+bi（0≤bi＜S/pi）$，
然后枚举$m$，$n=mS+bi*pi$，后面这个东西可以用个多重背包搞定，前面那个东西算方案就是个组合数

---

---

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const ll Mod = 1e9+7;
const int maxs = 2*1e6+1;
const int maxm = 7;
const int maxh = maxm*maxs;

int S,p[maxm+1],pr;
ll N[maxm+1];
ll n;
ll pw(ll x,int k)
{
    ll r=1; x%=Mod;
    for(ll t=x;k;t=t*t%Mod,k>>=1)
        if(k&1)r=r*t%Mod;
    return r;
}
ll C(ll n,ll m)
{
    n=n+1; m=m-1;
    n=n+m-1;
    ll ret=1;
    for(ll i=n;i>=n-m+1;i--)
        ret=ret*(i%Mod)%Mod;
    ret=ret*N[m]%Mod;
    return ret;
}
ll sum[maxs];
ll f[2][maxh];
int get_f()
{
    int now=0; f[0][0]=1;
    for(int i=1;i<=pr;i++)
    {
        now=1^now; int k=S/p[i]-1,up=i*S;
        for(int j=0;j<p[i];j++) sum[j]=0ll;
        for(int j=0,l=0;j<up;j++,l=(l+1)%p[i])
        {
            if(j/p[i]>k) 
                sum[l]-=f[1^now][j-(k+1)*p[i]];
            (sum[l]+=f[1^now][j])%=Mod;
            if(sum[l]<0) sum[l]+=Mod;
            f[now][j]=sum[l];
        }
        i=i+k*2-k*2;
    }
    return now;
}

int main()
{
    int t; scanf("%d%d",&S,&t); pr=0;
    int tmp=S;
    for(int i=2;i*i<=tmp;i++)if(tmp%i==0)
    {
        p[++pr]=i;
        tmp/=i;
        if(tmp%i==0)
        {
            while(t--) printf("0\n");
            return 0;
        }
    } if(tmp>1) p[++pr]=tmp;
    N[1]=1; for(ll i=2;i<=pr;i++) N[i]=N[i-1]*pw(i,Mod-2)%Mod;
    int now=get_f();
    while(t--)
    {
        scanf("%lld\n",&n);
        for(int i=1;i<=pr;i++) n-=p[i];
        if(n<0){printf("0\n");continue;}
        ll ret=0; ll up=(ll)n/S*S; ll k=pr*S;
        for(ll i=up;n-i<k&&i>=0;i-=S)
        {
            ll s1=f[now][n-i], s2=C(i/(ll)S,pr);
            ret+=s1*s2%Mod;
            ret%=Mod;
        }
        printf("%lld\n",ret);
    }

    return 0;
}