hdu 1398 Square Coins（母函數）

hdu 1398 Square Coins（母函數）

代碼：

#include
using namespace std;

int main()
{
    int n;
    int a[18];
    for(int i=1;i<=17;i++)
        a[i]=i*i;
    while(scanf("%d",&n)&&n)
    {
        long long c1[350],c2[350];
        for(int i=0;i<=n;i++)
        {
            c1[i]=1;
            c2[i]=0;
        }
        for(int i=2;a[i]<=n&&i<=17;i++)//判斷條件中i<=17不能省略，因為當n>289時，沒有退出循環的條件
        {
            for(int j=0;j<=n;j++)
            {
                for(int k=0;k+j<=n;k+=a[i])
                {
                    c2[j+k]+=c1[j];
                }
            }
            for(int j=0;j<=n;j++)
            {
                c1[j]=c2[j];
                c2[j]=0;
            }
        }
        printf("%lld\n",c1[n]);
    }
    return 0;
}