HDU-1398 Square Coins(生成函数)

题意

与$hdu1028$类似，只不过可用的数字都是平方数。

思路

类似的思路，注意下细节。

代码

#include <bits/stdc++.h>
#define DBG(x) cerr << #x << " = " << x << endl;

using namespace std;

const int N = 300 + 5;

int n, c[2][N];

int main() {
    while(~scanf("%d", &n) && n) {
        for(int i = 0; i <= n; i++) c[1][i] = 1, c[0][i] = 0;
        for(int i = 2; i <= n; i++) {
            for(int j = 0; j <= n; j++) {
                for(int k = 0; k + j <= n; k += i * i) c[i & 1][k + j] += c[1 - (i & 1)][j];
            }
            for(int j = 0; j <= n; j++) c[1 - (i & 1)][j] = 0;
        }
        printf("%d\n", c[n & 1][n]);
    }
    return 0;
}