hdu 1398 Square Coins(母函数)

#include <iostream>
#include<cstdio>
using namespace std;
// s(x)=(1+x+x^2+x^3+---+x^n)(1+x^4+x^16+---+x^n)---(1+x^289).
int c1[305],c2[305];
int a[18];
int main()
{
    int i,j,k;
    for(i=1;i<=17;i++)a[i]=i*i;
    for(i=0;i<=300;i++){
        c1[i]=1;
        c2[i]=0;
    }
    for(i=2;i<=17;i++){
        for(j=0;j<=300;j++){
            for(k=0;k+j<=300;k+=a[i]){
                c2[k+j]+=c1[j];
            }
        }
        for(j=0;j<=300;j++){
            c1[j]=c2[j];
            c2[j]=0;
        }
    }
    int n;
    while(cin>>n&&n){
        printf("%d\n",c1
);
    }
    return 0;
}