母函数模版详解和示例

                              母函数问题：

母函数，又称生成函数，是ACM竞赛中经常使用的一种解题算法，常用来解决组合方面的题目。

本文讲解母函数，但不讲解该算法的基础理论。读者随便找一本组合数学教材便可找到相应的内容，或者直接在网上搜索一下。

母函数通常解决类似如下的问题：

给5张1元，4张2元，3张5元，要得到15元，有多少种组合？

某些时候会规定至少使用3张1元、1张2元、0张5元。

某些时候会规定有无数张1元、2元、5元。

……

直接给出通用模板；

1. //为计算结果，b为中间结果。  
2. int a[MAX],b[MAX];  
3. //初始化a  
4. memset(a,0,sizeof(a));  
5. a[0]=1;  
6. for (int i=1;i<=17;i++)//循环每个因子  
7. {  
8.     memset(b,0,sizeof(b));  
9.     for (int j=n1[i];j<=n2[i]&&j*v[i]<=P;j++)//循环每个因子的每一项  
10.         for (int k=0;k+j*v[i]<=P;k++)//循环a的每个项  
11.             b[k+j*v[i]]+=a[k];//把结果加到对应位  
12.     memcpy(a,b,sizeof(b));//b赋值给a  
13. }  （转载至某博客）
14. P是可能的最大指数。拿钞票组合这题来说，如果要求15元有多少组合，那么P就是15；如果问最小的不能拼出的数值，那么P就是所有钱加起来的和。P有时可以直接省略。具体请看本文后面给出的例题。
    如果 n2是无穷，那么第二层循环条件j<=n2[i]可以去掉。

15. 如何提高效率？

    用一个last变量记录目前最大的指数，这样只需要在0..last上进行计算。

    这里给出第二个模板：

    [cpp]  view plain  copy 
    

    1. //初始化a，因为有last，所以这里无需初始化其他位  
    2. a[0]=1;  
    3. int last=0;  
    4. for (int i=0;i<K;i++)  
    5. {  
    6.     int last2=min(last+n[i]*v[i],P);//计算下一次的last  
    7.     memset(b,0,sizeof(int)*(last2+1));//只清空b[0..last2]  
    8.     for (int j=n1[i];j<=n2[i]&&j*v[i]<=last2;j++)//这里是last2  
    9.         for (int k=0;k<=last&&k+j*v[i]<=last2;k++)//这里一个是last，一个是last2  
    10.             b[k+j*v[i]]+=a[k];  
    11.     memcpy(a,b,sizeof(int)*(last2+1));//b赋值给a，只赋值0..last2  
    12.     last=last2;//更新last  
    13. }  

          事实上，可以通过不设置初始数组的方式，直接输入数值，然后用数值进行循环的大小判断。

          例题如下，

又到了选课的时间了，xhd看着选课表发呆，为了想让下一学期好过点，他想知道学n个学分共有多少组合。你来帮帮他吧。（xhd认为一样学分的课没区别） 

Input

输入数据的第一行是一个数据T，表示有T组数据。 
每组数据的第一行是两个整数n(1 <= n <= 40)，k(1 <= k <= 8)。 
接着有k行，每行有两个整数a(1 <= a <= 8),b(1 <= b <= 10)，表示学分为a的课有b门。 

Output

对于每组输入数据，输出一个整数，表示学n个学分的组合数。 

Sample Input

2
2 2
1 2
2 1
40 8
1 1
2 2
3 2
4 2
5 8
6 9
7 6
8 8

Sample Output

2
445

代码如下：

另外，至于什么时候用第一个模板，什么时候用第二个模板，就看题目规模。

通常情况下，第一个模板就够用了，上面的那些用第二个模板的题目用第一个模板同样能AC。

但如果数据规模比较大（通常不会有这种情况），就要使用第二个模板了。

以上题目n1均为0。

例题：

假设有x1个字母A， x2个字母B,..... x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,..... 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词呢？单词的价值就是组成一个单词的所有字母的价值之和，比如，单词ACM的价值是1+3+14=18，单词HDU的价值是8+4+21=33。(组成的单词与排列顺序无关，比如ACM与CMA认为是同一个单词）。

Input

输入首先是一个整数N，代表测试实例的个数。 
然后包括N行数据，每行包括26个<=20的整数x1,x2,.....x26. 

Output

对于每个测试实例，请输出能找到的总价值<=50的单词数,每个实例的输出占一行。

Sample Input

2
1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
9 2 6 2 10 2 2 5 6 1 0 2 7 0 2 2 7 5 10 6 10 2 10 6 1 9

Sample Output

7

379297

代码如下：

对于该代码，首先内部三层循环，第一层循环是代表因数的项数，

第二层循环代表价值从0到一，从0开始是为了避免后面的下一层循环丢失因数，导致数据丢失，

第三层循环是最重要的循环，中间判断条件是在物品价值或者其他参数范围内，并且所求参数小于题目限定的最大参数值，这个根据题目条件设置，可以自由选择，另一种写法如下：

#include <iostream>

using namespace std;

int main()

{

        int N;

        while(cin>>N)

               while (N--)

               {

                       int wordvalue,temp[55],zuhe[55];

                       memset(zuhe, 0, sizeof(zuhe));

                       memset(temp, 0, sizeof(temp));

                       cin >> wordvalue;

                       for (int i = 0; i <= wordvalue; i++)

                              zuhe[i] = 1;

                       for (int i = 2; i <= 26; i++)

                       {

                              cin >> wordvalue;

                              if (!wordvalue)continue;

                              for(int j=0;j<=50;j++)

                                      for (int k = 0; k*i + j <= 50 && k <= wordvalue; k++)

                                      {

                                             temp[k*i + j] +=zuhe[j];

                                      }

                              for (int i = 0; i <= 50; i++)

                              {

                                      zuhe[i] = temp[i];

                                      temp[i] = 0;

                              }

                       }

                       int s = 0;

                       for (int i = 1; i <= 50; i++)

                              s += zuhe[i];

                       cout << s << endl;

               }

        return 0;

}