组合数学——生成函数
参考:
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;
}