HDOJ 1398 生成函数

链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1398

题意：

给你17种硬币，面值分别为1²，2²，3²，4²……

询问组成n元有多少种方法

代码：

int n;
int v[20];
int a[310], b[310], last;

int main() {
    rep(i, 0, 18) v[i] = i*i;
    a[0] = 1;
    rep(i, 1, 18) {
        memset(b, 0, sizeof(b));
        for (int j = 0; j*v[i] < 310; j++)
            for (int k = 0; k + j*v[i] < 310; k++)
                b[k + j*v[i]] += a[k];
        memcpy(a, b, sizeof(b));
    }
    while (cin >> n, n) cout << a[n] << endl;
    return 0;
}