hdu 1398 Square Coins (母函数)

题意:有面值为平方数的硬币若干,最小为1,最大为17^2,给出面值n,问能有多少种组成方案。


生成函数如下:

G(x)=(1+x+x2+x3+x4+…)(1+x4+x8+x12+…)(1+x9+x18+x27+…)…



#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int c[305],c0[305];
void init(int n)
{
    int i,j,k;
    for(i=0;i<=n;++i) c0[i]=1,c[i]=0;
    for(k=2;k<=17;++k)
    {
        for(i=0;i<=n;++i)
            for(j=0;i+j<=n;j+=k*k)
                    c[i+j]+=c0[i];
        for(i=0;i<=n;++i) c0[i]=c[i],c[i]=0;
    }
}
int main()
{
    int n;
    init(300);
    while(~scanf("%d",&n)&&n>0)
        printf("%d\n",c0[n]);
    return 0;
}



你可能感兴趣的:(hdu 1398 Square Coins (母函数))