hdu 1398 Square Coins【生成函数】

预处理出完全平方数就和普通的生成函数解整数拆分一样了

#include<iostream>
#include<cstdio>
using namespace std;
const int N=605;
int n,m,q[N],a[N],b[N];
int main()
{
	for(int i=1;i<=18;i++)
		q[i]=i*i;
	while(scanf("%d",&m)&&m)
	{
		for(n=1;q[n+1]<=m;n++);
		for(int i=0;i<=m;i++)
			a[i]=1,b[i]=0;
		for(int i=2;i<=n;i++)
		{
			for(int j=0;j<=m;j++)
				for(int k=0;k<=m;k+=q[i])
					b[j+k]+=a[j];
			for(int j=0;j<=m;j++)
				a[j]=b[j],b[j]=0;
		}
		printf("%d\n",a[m]);
	}
	return 0;
}