母函数及相关的算法题
母函数即生成函数,构造这么一个多项式函数g(x),使得x的n次方系数为f(n),是组合数学中尤其是计数方面的一个重要理论和工具。
(1+a1x)(1+a2x)(1+a3x)...(1+anx)=1+(a1+a2+a3+...+an)x+(a1a2+a1a3+...+an-1an)x2+...+(a1a2a3*...*an)xn
由此可以看出:
1. x的系数是a1,a2,…an的单个组合的全体。相当于从a1,a2,…an选1个进行组合,然后加在一起
2. x2的系数是a1,a2,…an的两个组合的全体。相当于从a1,a2,…an选2个进行组合,然后加在一起
………
n. xn的系数是a1,a2,….an的n个组合的全体(只有1个)。相当于从a1,a2,…an选n个进行组合,然后加在一起
如果我们把a1,a2,…an的值设为1的话
原式就变成了
(1+x)n=1+C(n,1)x+C(n,2)x2+C(n,3)x3+…+C(n,n)xn
利用这种关系,我们可以解决一些算法问题
例1、有面值为1,2,3的硬币各一个,问有几种组合?
我们把所有的组合这么想:
(不拿面值为1的,拿一个面值为1的)*(不拿一个面值为2的,拿一个面值为2的)*(不拿一个面值为3的,拿一个面值为3的)
==(面值为0的个数,面值为1的个数,面值为2的个数,面值为3的个数,面值为4的个数,面值为5的个数,面值为6的个数)
我们用x表示硬币,x的指数表示硬币的面值,这样:
1克的硬币组合可以用函数x0+x1表示,x0表示不拿一个面值为1的,x1表示拿一个面值为1的
2克的硬币组合可以用函数x0+x2表示,x0表示不拿一个面值为2的,x2表示拿一个面值为2的
3克的硬币组合可以用函数x0+x3表示,x0表示不拿一个面值为3的,x3表示拿一个面值为3的
所有的组合就可以这么表示出来了:
(1+x)(1+x2)(1+x3)==1+x+x2+2x3+x4+x5+x6
x前的系数就相当于称出相应硬币的面值的组合数目
比如面值为3的有两种组合,(拿一个面值为3的,和那一个1个面值为1和面值为2的)
例2、有面值为1的硬币3个,面值为2的硬币2个,面值为3的硬币1个,问有多少种组合?
1克的硬币组合可以用函数x0+x1 +x2+x3表示,x0表示不拿一个面值为1的,x1表示拿1个面值为1的,x2表示拿2个面值为1的,x3表示拿3个面值为1的
2克的硬币组合可以用函数x0+x2+x4表示,x0表示不拿一个面值为2的,x2表示拿一个面值为2的,x4表示拿3个面值为2的
3克的硬币组合可以用函数x0+x3表示,x0表示不拿一个面值为3的,x3表示拿1个面值为3的,x6表示拿2个面值为3的
所有的组合就可以这么表示出来了:
(x0+x1 +x2+x3)*(x0+x2+x4)*( x0+x3)
例3、有面值为1,2,3的硬币无数个,问有多少种组合?
1克的硬币组合可以用函数x0+x1 +x2+…+xn表示,分别代表拿0个,1个,2个…n个的情况。
2克的硬币组合可以用函数x0+x2+x4+…+x2n表示,分别代表拿0个,1个,2个,n个的情况。
3克的硬币组合可以用函数x0+x3+x6+…+x3n表示,分别代表拿0个,1个,2个,n个的情况。
所有的组合就可以这么表示出来了:
(x0+x1 +x2+…+xn)*(x0+x2+x4+…+x2n)*(x0+x3+x6+…+x3n)
现在就例2的例子给出代码与分析
有面值为1的硬币3个,面值为2的硬币2个,面值为3的硬币1个,问有多少种组合?
//(x0+x1 +x2+x3)*(x0+x2+x4)*( x0+x3) struct Coin { int value; int number; }; const int MAX = 50; int main() { Coin coin[3]; coin[0].value = 1; coin[0].number = 3; coin[1].value = 2; coin[1].number = 2; coin[2].value = 3; coin[2].number = 1; int result[MAX] = { 0 }; int temp[MAX] = { 0 }; int number; while (cin >> number) { for (int i = 0; i < MAX; i++) { result[i] = 0; temp[i] = 0; } for (int i = 0; i <=coin[0].number;i++) { result[i*coin[0].value] = 1; } for (int i = 1; i < 3; i++) { for (int j = 0; j <=number; j++) { for (int k = 0, index = 0; index <=coin[i].number && (k + j)<=number; index++, k += coin[i].value) { temp[j + k] += result[j]; } } for (int j = 0; j <=number; j++) { result[j] = temp[j]; temp[j] = 0; } } cout << result[number] << endl; } }
我们对②进行分析
它的循环不变式是result[0]-result[number]记录了每次新增面额为coin[i].value时的所有组合数目,面值为i的有result[i]种组合。
初始化:首先,先来证明在第一轮迭代之前,它是成立的。 在①中,把第一个面额硬币的组合赋给了result,for (int i = 0; i <=coin[0].number;i++)①
{
result[i*coin[0].value] = 1;
}
此时result记录了第一个面额硬币的所有组合及其数目。所以在循环开始前循环不变式是正确的。
保持: 这里的i代表了当前计算的式子,因为①已经计算完了第一个式子,所以i从1开始,到2截止(数组的下标是从0开始)
j每次需要更新的面值,所以从0开始到number截止,
k代表了第i个式子的增量,index第i个式子的第几个遍历。所以对于第i个式子,k为coin[i].value,index从0开始到coin[i].number截止;当index已经coin[i].number或是当前面值j加上增量k之后已经超过了所要求的面值即可停止循环
每次迭代面值为j的,现在面值变成了j+k的,所以面值为j+k的为为原有的面值为j的数目加上新的面值为j+k的数目
然后将temp赋予result,将temp清零,开始下一次迭代。
终止:对此算法来说,当i大于数组的长度(即所有面额的硬币都计算完毕),for循环结束。这时result[0]-result[number]记录了面额为0到number所有组合的数目
参考文章:
母函数(Generating function)详解 http://www.wutianqi.com/?p=596
什么是生成函数? http://www.matrix67.com/blog/archives/120
posted on 2014-11-28 21:25 magicsoar 阅读(2055) 评论(0) 编辑 收藏 举报