hdu 4658 Integer Partition

http://acm.hdu.edu.cn/showproblem.php?pid=4658

这道题不会做，看着别人的写的。这道题求的是一个数的划分，但是划分中一个数不能重复k次，用到了五边形数定理，没看懂。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100005
using namespace std;
const int mod=1000000007;

int dp[maxn];

void inti()//一个数的划分
{
    dp[0]=1;
    for(int i=1; i<=100000; i++)
    {
        dp[i]=0;
        for(int j=1; ; j++)
        {
            int x=(3*j-1)*j/2;
            if(x>i) break;
            int s=dp[i-x];
            if(x+j<=i) s=(s+dp[i-j-x])%mod;
            if(j&1) dp[i]=(dp[i]+s)%mod;
            else dp[i]=(dp[i]-s+mod)%mod;
        }
    }
}

int main()
{
    inti();
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        int ans=dp[n];
        for(int i=1; ; i++)
        {
            int x=k*i*(3*i-1)/2;
            if(x>n) break;
            int c=dp[n-x];
            if(x+i*k<=n) c=(c+dp[n-x-k*i])%mod;
            if(i&1) ans=(ans-c+mod)%mod;
            else ans=(ans+c)%mod;
        }
        printf("%d
",ans);
    }
    return 0;
}