hdu 4658 整数划分+五边形定理 多校第六场

首先要知道五边形定理是什么东西

它揭示了整数划分的母函数是欧拉函数的倒数，欧拉函数可以通过级数表示

P(x)*phi(x)=1  -> p(n)-p(n-1)-p(n-2)+p(n-5)+p(n-7)......=0

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define N 100000
#define mod 1000000007
#define ll long long
ll p[N+10];

void init()
{
    p[0]=p[1]=1;
    for(int i=2;i<=N;++i)
    {
        p[i]=0;int x=1;
        for(int k=1,j=1;x>=0;++k,j*=-1)
        {
            x=i-(3*k*k+k)/2;
            if(x>=0) p[i]+=p[x]*j;
            x=i-(3*k*k-k)/2;
            if(x>=0) p[i]+=p[x]*j;
            if(p[i]>=mod||p[i]<0)
                p[i]=(p[i]%mod+mod)%mod;
        }
    }
}

int main ()
{
    init();
    int T;scanf("%d",&T);
    while(T--)
    {
        int n,K;
        scanf("%d%d",&n,&K);
        ll ans=p[n];
        int x=1;
        for(int k=1,j=-1;x>=0;++k,j*=-1)
        {
            x=n-(3*k*k+k)/2*K;
            if(x>=0) ans+=p[x]*j;
            x=n-(3*k*k-k)/2*K;
            if(x>=0) ans+=p[x]*j;
            if(ans>=mod||ans<0)
                ans=(ans%mod+mod)%mod;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}