题意
输入一个n,求n的分拆平方数,例如:
10=1+1+1+1+1+1+1+1+1+1
10=4+1+1+1+1+1+1
10=4+4+1+1
10=9+1
因此10有四种组合,所以输入10时输出4。
思路
母函数裸题,与hdu1028相似,只不过那里是求n的分拆数,这里是分拆平方数,所以只要把i改为i*i即可。
#include<iostream>
using namespace std;
const int maxn=300;
int c1[maxn+1],c2[maxn+1];
int main()
{
int i,j,k,n;
for(i=0;i<=maxn;i++)
{
c1[i]=1;
c2[i]=0;
}
for(i=2;i*i<=maxn;i++)
{
for(j=0;j<=maxn;j++)
for(k=0;k+j<=maxn;k+=i*i)
c2[j+k]+=c1[j];
for(j=0;j<=maxn;j++)
{
c1[j]=c2[j];
c2[j]=0;
}
}
while(cin>>n&&n)
cout<<c1[n]<<endl;
return 0;
}