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;
}