HDU1398(生成函数)

有1,4,9,16,25…..2^17这17种面值的硬币，问任意给定一个不大于300的正整数面额，用这些硬币来组成此面额总共有多少种组合种数。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxv=25;
const int maxn=300+5;
int v[maxv];
int n1[maxn];
int n;
int a[maxn],b[maxn];
//a为计算结果，b为中间量
void init(int P)
{
    //计算权值
    for(int i=1; i<=17; i++)
    {
        v[i]=i*i;
    }
    memset(a,0,sizeof(a));
    memset(n1,0,sizeof(n1));
    a[0]=1;
    for (int i=1; i<=17; i++) //遍历每种因子
    {
        memset(b,0,sizeof(b));
        for (int j=n1[i]; j*v[i]<=P; j++) //遍历每种因子所有可能的数量
            for (int k=0; k+j*v[i]<=P; k++) //遍历a的每个项
                b[k+j*v[i]]+=a[k];//计算相应的系数
        memcpy(a,b,sizeof(b));
    }
}
int main()
{

    init(300);
    while(scanf("%d",&n)&&n)
    {
        printf("%d\n",b[n]);
    }
    return 0;
}