hdu 1398 整数划分变形 （母函数）

有1,4,9,16,25.....2^17这么多面值的硬币，问任意给定一个不大于300的正整数面额，用这些硬币来组成此面额总共有多少种组合种数

比如10
全1
4 + 6个 1
4+4+1+1
9+1

求(1+x+x^2+x^3+x^4+x^5+x^6+x^7+x^8+x^9+x^10)(1+x^4+x^8)(1+x^9)中
x^10的系数即可

只是把模板的 i<=n改成了i*i<=n,
在k遍历指数时把k=k+i变成了k=k+i*i

Sample Input
2
10
30
0

Sample Output
1
4
27

 

# include <iostream>
# include <cstdio>
# include <cstring>
# include <algorithm>
# include <string>
# include <cmath>
# include <queue>
# include <list>
# define LL long long
using namespace std ;

int c1[10010], c2[10010] ;
int main()
{
    //freopen("in.txt","r",stdin) ;
    int n;
    int i, j, k;
    while(cin >> n)
    {
        if (n == 0)
            break ;
        for(i=0; i<=n; ++i)
        {
            c1[i] = 1;
            c2[i] = 0;
        }
        for(i=2; i*i<=n; ++i)
        {

            for(j=0; j<=n; ++j)
                for(k=0; k+j<=n; k += i*i)
                {
                    c2[j+k] += c1[j];
                }
                for(j=0; j<=n; ++j)
                {
                    c1[j] = c2[j];
                    c2[j] = 0;
                }
        }
        cout << c1[n] << endl;
    }
    return 0;
}