生成函数-HDU1398

https://vjudge.net/contest/181055#problem/A

给你一个数 让你求出1~17中每个数的平方任意组合 有几种情况满足条件。

有多少种满足方案输出a[n] 最小的取不到的面值输出i，此时a[i]为0且i最小

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <queue>
#define ll long long
using namespace std;
const int maxn=1e5+10;
const int mod=1000000007;
int a[10000],b[10000];
int v[20];
int main()
{
    int P;
    for(int i=1;i<=17;i++)
        v[i]=i*i;
    while(~scanf("%d",&P),P)
    {
        memset(a,0,sizeof(a));
        a[0]=1;
        for (int i=1; i<=17; i++) //循环每个因子
        {
            memset(b,0,sizeof(b));
            for (int j=0; j*v[i]<=P; j++) //循环每个因子的每一项
                for (int k=0; k+j*v[i]<=P; k++) //循环a的每个项
                    b[k+j*v[i]]+=a[k];//把结果加到对应位
            memcpy(a,b,sizeof(b));//b赋值给a
        }
        cout<<a[P]<<endl;
    }
    return 0;
}