hdu 1394 母函数入门题

刘春英老师的母函数ppt：母函数PPT

这里写出母函数：（1+x+x^2+x^3+x^4…+x^n）(1+x^4+x^8+…x^k（k<=n）)（1+x^9+x^18+…+x^（k<=n））…

展开：就是从左往右展开，两项两项乘起来，得到结果后再往右乘，a[i]保存的就是结果的x^i的系数，b[i]则是临时变量

#include<bits/stdc++.h>using namespace std;#define mod 1000000007#define PI acos(-1.0)#define INF 0x3f3f3f3ftypedef long long LL;typedef unsigned long long ULL;const int maxn=310;LL a[maxn],b[maxn];int n;int main(){    while(~scanf("%d",&n)){        if(!n)break;        for(int i=0;i<=n;i++){            a[i]=1;            b[i]=0;        }        for(int i=2;i<=17;i++){            for(int j=0;j<=n;j++){                for(int k=0;k+j<=n;k+=i*i){                    b[k+j]+=a[j];                }            }            for(int j=0;j<=n;j++){                a[j]=b[j];                b[j]=0;            }        }        printf("%lld\n",a[n]);    }    return 0;}