hdu 1398 Square Coins （母函数）

#include<stdio.h>

#include<string.h>

int main()

{

    int i,j,k,n,a[18][301];

    memset(a,0,sizeof(a));

    for(i=0;i<301;i++)

        a[1][i]=1;

    for(i=2;i<18;i++)

    {

        for(j=0;j<301;j++)

            for(k=0;j+i*i*k<301;k++)

                a[i][j+i*i*k]+=a[i-1][j];

    }

    while(scanf("%d",&n)==1 && n)

    {

        printf("%d\n",a[17][n]);

    }

    return 0;

}