母函数讲解:https://blog.csdn.net/ftx456789/article/details/79084138
母函数核心代码:
for(int i=0;i<n;i++)
{
for(int j=0;j<maxn;j++)
for(int k=0;k<=a[i];k++)
c2[j+value[i]*k]+=c1[j]
memcpy(c1,c2,sizeof(c1));
memset(c2,0,sizeof(c2));
}
第一重循环的i代表这里的n个多项式,所以约束条件就是i<ni<n,然后第二重循环的j代表前i个已经乘完的多项式的结果,大于当前最大次幂的部分的系数反正为0,约束条件是最后的多项式的最大幂次,所以约束条件就是j<maxnj<maxn,第三重循环k代表下一个加入乘法的多项式,约束条件就是这个多项的个数
还是我们上面的砝码问题,程序模拟的话n=3,maxn就是这些砝码所能构成的最大的重量,然后a[i]储存的就是每一种砝码的数量,value储存的就是每种砝码的价值也就是重量
次幂的乘法会把原来的系数也带上,而次幂的乘法就是次幂与次幂之间的和,c1和c2就是代表次幂,里面存的值也就是方法数
//普通母函数应用
#include<iostream>
#include<cstring>
using namespace std;
int main()
{
//c1 存储多项式的系数 c2临时数组,保存中间临时情况
int c1[300],c2[300];
int n;
while(cin>>n)
{
//初始化第一个多项式
//注意多项式是从x0开始,直到 xn ,所以 i最后为 <= n;
for(int i=0;i<=n;i++)
{
c1[i]=1;
c2[i]=0;
}
for(int i=2;i<=n;i++)
{
// 遍历当前结果多项式的每一项(当前结果的第j项)与第i个多项式相乘,
for(int j=0;j<=n;j++)
{
// 遍历第i个多项式的每一项,此处构造用小砝码组成大砝码的多项式
//注意此处 k 的变化为 k+=i
for(int k=0;k<=n;k+=i)
{
//幂运算
c2[j+k]+=c1[j];
}
}
// 将临时的结果覆盖当前结果,同时把临时结果置零,为下次做准备
memcpy(c1,c2,sizeof(c1));
memset(c2,0,sizeof(c2));
}
cout<<c1[n]<<endl;
}
return 0;
}
HDU 1521(求排列数)
//指数型母函数
#include<iostream>
#include<cstring>
#include<stdio.h>
using namespace std;
double c1[20],c2[20];
int num[20];
double jc[20];
int main()
{
jc[0]=jc[1]=1;
//计算阶乘
for(int i=1;i<=10;i++)
{
jc[i]=jc[i-1]*i;
}
int n,m;
while(cin>>n>>m)
{
for(int i=0;i<n;i++)
{
cin>>num[i];
}
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
c1[0]=1;
for(int i=0;i<=n;i++)
{
for(int j=0;j<=m;j++)
{
//注意 k 的取值范围,k 要小于它的数量以及 j+k要小于要选出的数量数
for(int k=0;k<=num[i]&&j+k<=m;k++)
{
c2[j+k]+=c1[j]/jc[k];
}
}
memcpy(c1,c2,sizeof(c1));
memset(c2,0,sizeof(c2));
}
//输出浮点数的整数部分,省略全部小数部分(以及小数点)
//注意直接cout 会WA
printf("%.0lf\n",c1[m]*jc[m]);
}
return 0;
}