母函数 学习笔记

组合数学——生成函数

参考：

1. http://www.wutianqi.com/blog/596.html

2. https://www.cnblogs.com/linyujun/p/5207730.html 

个人理解：

用多项式乘法表示某种组合的可能的次数。

 

 

模版1：

#include 
using namespace std;
// Author: Tanky Woo
// www.wutianqi.com
const int _max = 10001; 
// c1是保存各项质量砝码可以组合的数目
// c2是中间量，保存没一次的情况
int c1[_max], c2[_max];   
int main()
{   //int n,i,j,k;
    int nNum;   // 
    int i, j, k;

    while(cin >> nNum)
    {
        for(i=0; i<=nNum; ++i)   // ---- ①
        {
            c1[i] = 1;
            c2[i] = 0;
        }
        for(i=2; i<=nNum; ++i)   // ----- ②
        {

            for(j=0; j<=nNum; ++j)   // ----- ③
                for(k=0; k+j<=nNum; k+=i)  // ---- ④
                {
                    c2[j+k] += c1[j];
                }
            for(j=0; j<=nNum; ++j)     // ---- ⑤
            {
                c1[j] = c2[j];
                c2[j] = 0;
            }
        }
        cout << c1[nNum] << endl;
    }
    return 0;
}

/*
> > ① 、首先对c1初始化，由第一个表达式(1+x+x^2+..x^n)初始化，把质量从0到n的所有砝码都初始化为1. > > > > ② 、 i从2到n遍历，这里i就是指第i个表达式，上面给出的第二种母函数关系式里，每一个括号括起来的就是一个表达式。 > > > > ③、j 从0到n遍历，这里j就是(前面i個表达式累乘的表达式)里第j个变量，(这里感谢一下seagg朋友给我指出的错误，大家可以看下留言处的讨论)。如(1+x)(1+x^2)(1+x^3)，j先指示的是1和x的系数，i=2执行完之后变为 （1+x+x^2+x^3）(1+x^3)，这时候j应该指示的是合并后的第一个括号的四个变量的系数。 > > > > ④ 、 k表示的是第j个指数，所以k每次增i（因为第i个表达式的增量是i）。 > > > > ⑤ 、把c2的值赋给c1,而把c2初始化为0，因为c2每次是从一个表达式中开始的。 > >
*/

模版2:

#include<cstdio>
typedef long long LL;
const int N = 100 + 5;//假如题目只问到100为止 
const int MAX = 3;//题目只有1,2,3这3种邮票 
LL c1[N], c2[N];//c2是临时合并的多项式，c1是最终合并的多项式 
int n;
void init(){
    c1[0] = 1;//一开始0的情况算一种 
    for(int i = 1; i <= MAX; i ++){//把1分到MAXN的邮票合并，变成一个多项式 
        for(int j = 0; j < N; j += i){//i分的邮票，步长是i
            for(int k = 0; j + k < N; k ++){//从x^0到x^N遍历一遍 
                c2[j + k] += c1[k];//因为j的所有项系数为1，所以c1[k]可以看成c1[k]*1; 
            }
        } 
        for(int j = 0; j < N; j ++){//把c2的数据抄到c1，清空c2 
            c1[j] = c2[j];
            c2[j] = 0;
        }
    }
} 
int main(){
    init();
    while(scanf("%d", &n) != EOF){
        printf("%I64d\n", c1[n]);
    }
}

 

 

题目1:

HDU1028 Ignatius and the Princess III

http://acm.hdu.edu.cn/showproblem.php?pid=1028

#include <iostream>
#define maxn 125
using namespace std;
typedef long long ll;
ll c1[maxn], c2[maxn];
void init() {
    for(int i = 0; i <= maxn; i++) {
        c1[i] = 1;
        c2[i] = 0;
    }
    for(int i = 2; i <= maxn; i++) {
        for(int j = 0; j <= maxn; j++) {
            for(int k = 0; k+j < maxn; k+=i) {
                c2[j + k] += c1[j];
            }
        }
        for(int j = 0; j <= maxn; j++) {
            c1[j] = c2[j];
            c2[j] = 0;
        }
    }
}
int main() {
    int N;
    while(cin >> N) {
        init();
        cout << c1[N] << endl;
    }
    
    return 0;
}

 

题目2:

HDU 1398 Square Coins

http://acm.hdu.edu.cn/showproblem.php?pid=1398

AC代码：

#include <iostream>
#define maxn 300 + 50
using namespace std;
typedef long long ll;
ll c1[maxn], c2[maxn];
void init() {
    c1[0] = 1;
    for(int i = 1; i <= 17; i++) {
        for(int j = 0; j < maxn; j += i * i) {
            for(int k = 0; j + k < maxn; k++) {
                c2[j + k] += c1[k];
            }
        }
        for(int j = 0; j < maxn; j++) {
            c1[j] = c2[j];
            c2[j] = 0;
        }
    }
}
int main() {
    int N;
    init();
    while(cin >> N) {
        if(N == 0) return 0;
        cout << c1[N] << endl;
    }

    return 0;
}

 

题目3:

HDU 1085 Holding Bin-Laden Captive!

http://acm.hdu.edu.cn/showproblem.php?pid=1085

 

AC代码：

#include <iostream>
#include <cstring>
#define maxn 8000 + 50
using namespace std;
typedef long long ll;
ll c1[maxn], c2[maxn];
int cost[3] = {1, 2, 5};
int num[3];
void init(int a, int b, int c) {
    memset(c1, 0, sizeof c1);
    memset(c2, 0, sizeof c2);
    num[0] = a, num[1] = b, num[2] = c;
    int max = a + 2 * b + 5 * c;
    c1[0] = 1;
    for(int i = 0; i < 3; i++) {
        for(int j = 0; j <= num[i] * cost[i]; j += cost[i]) {
            for(int k = 0; j + k < maxn; k++) {
                c2[j + k] += c1[k];
            }
        }
        for(int j = 0; j < maxn; j++) {
            c1[j] = c2[j];
            c2[j] = 0;
        }
    }
}
int main() {
    int a, b, c;
    while(cin >> a >> b >> c) {
        if(a == 0 && b == 0 && c == 0) return 0;
        else {
            init(a, b, c);
            for (int i = 1; i < maxn; i++) {
                if (!c1[i]) {
                    cout << i << endl;
                    break;
                }
            }
        }
    }

    return 0;
}