母函数

这里是Tanky Woo对母函数的详细讲解:http://http://www.wutianqi.com/?p=596

Orz......

贴一道稍简单,稍经典的题, 备忘:

题目链接:http://http://acm.hdu.edu.cn/showproblem.php?pid=1171

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 250000 + 5;
typedef long long LL;

int c1[maxn], c2[maxn];
int num[55], val[55];
int main()
{
	freopen("f:\\input.txt", "r", stdin);
	int i, j, k;
	int N;
	while ((cin >> N) && N > 0)
	{
		int sum = 0;
		for (i = 1; i <= N; i++)
		{
			cin >> val[i] >> num[i];
			sum += val[i] * num[i];
		}

		for (i = 0; i <= sum; i++)
		{
			c1[i] = (i%val[1] == 0 &&i<=val[1]*num[1]? 1 : 0);
			c2[i] = 0;
		}
		int pre_sum = 0;
		for (i = 2; i <= N; i++)//第i个表达式
		{
			pre_sum += val[i - 1] * num[i - 1];//(当前c1[]中指数的最大值)
			for (j = 0; j <= pre_sum;j++)//取出第一个括号里的指数为j的项
			if (c1[j])
			for (k = 0; k / val[i] <= num[i]; k += val[i])//更新指数为j+k的项(指数为j的项*指数为k的项)
			{
				c2[j + k] += c1[j];
			}

			for (j = 0; j <= pre_sum+val[i]*num[i]; j++)//将c2[]中的结果保存到c1[]中
			{					//(pre_sum+val[i]*num[i]为当前c2[]中指数的最大值)
				c1[j] = c2[j];
				c2[j] = 0;
			}
		}
		for (i = (sum - 1) / 2 + 1; i <= sum;i++)
		if (c1[i])
			break;
		cout << i << " " << sum - i << endl;
	}
	return 0;
}

最后使用了下面的作为模板:

/*题目:求用1分、2分、3分的邮票(每种邮票可以选无限张)贴出不同数值的方案数：
G(x)=(1+x+x^2+x^3...)(1+x^2+x^4...)(1+x^3+x^6...)
*/
#include <iostream>
using namespace std;
const int _max = 10001;
// c1是保存各项质量砝码可以组合的数目
// c2是中间量，保存没一次的情况
int c1[_max], c2[_max];
int main()
{	
	int nNum;   
	int i, j, k;

	while (cin >> nNum)
	{
		for (i = 0; i <= nNum; ++i)   // 将c1[]初始化为第一个表达式---- ①即(1+x+x^2+..x^n)的初始化.
		{
			c1[i] = 1;
			c2[i] = 0;
		}
		for (i = 2; i <= 3; ++i)   // 第i个表达式----- ②
		{
			for (j = 0; j <= nNum; ++j) // 取出第一个括号里的指数为j的项(一些题或许加上条件判断if(c1[j])更省时)----- ③。
			for (k = 0; k + j <= nNum; k += i)  //对每个j遍历k,更新指数为j+k的项(即指数为j的项*指数为k的项)---- ④
			{
				c2[j + k] += c1[j];
			}
			for (j = 0; j <= nNum; ++j)  // ---- ⑤把c2[]的结果转移到c1[],把c2清零。
			{
				c1[j] = c2[j];
				c2[j] = 0;
			}
		}
		cout << c1[nNum] << endl;
	}
	return 0;
}