HDU 1028 Ignatius and the Princess III

HDU_1028

    这个题目可以用dp去解，用f[i][j]表示取到i这个数时和为j的方案数。

    为了练一下生成函数，我就用生成函数的思路写了一下，构造生成函数G(x)=(1+x+x^2+…)(1+x^2+x^4+…)…(1+x^p+x^(2*p)+…)…，然后每次查询相当于查询x^n的系数是多少。

#include<stdio.h>
#include<string.h>
#define MAXD 130
long long int f[MAXD][MAXD];
int N;
void prepare()
{
    int i, j, k;
    memset(f, 0, sizeof(f));
    f[0][0] = 1;
    for(i = 1; i <= 120; i ++)
        for(j = 0; j <= 120; j += i)
            for(k = 0; k + j <= 120; k ++)
                f[i][k + j] += f[i - 1][k];
}
int main()
{
    prepare();
    while(scanf("%d", &N) == 1)
        printf("%I64d\n", f[N][N]);
    return 0;
}