题意:http://acm.hdu.edu.cn/showproblem.php?pid=1398
母函数的基本应用,算作练习吧。
#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[n]);
}
return 0;
}