链接:http://acm.hdu.edu.cn/showproblem.php?pid=1028
题目:
Ignatius and the Princess III
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 12213 Accepted Submission(s): 8630
"The second problem is, given an positive integer N, we define an equation like this:
N=a[1]+a[2]+a[3]+...+a[m];
a[i]>0,1<=m<=N;
My question is how many different equations you can find for a given N.
For example, assume N is 4, we can find:
4 = 4;
4 = 3 + 1;
4 = 2 + 2;
4 = 2 + 1 + 1;
4 = 1 + 1 + 1 + 1;
so the result is 5 when N is 4. Note that "4 = 3 + 1" and "4 = 1 + 3" is the same in this problem. Now, you do it!"
4 10 20
5 42 627
这是一道典型的母函数(http://zh.wikipedia.org/wiki/%E6%AF%8D%E5%87%BD%E6%95%B0)的题目。
我们先用一个引例:
例1:
给定3个砝码,分别是1g,2g,3g,问:这3个砝码可以称量出那些质量,对于一个可以称量的质量有几种不同的称法,对于这个问题,我们可以这样解决,
对1g的砝码,给出一个方程 1 + x;(1 * x ^ 0 + 1 * x ^ 1,取1个0g的砝码,或是取1个1g的砝码,即1g的砝码取或不取);
对2g的砝码,给出一个方程 1 + x^2;(1 * x ^ 0 + 1 * x ^ 2,取1个0g的砝码,或是取1个2g的砝码,即2g的砝码取或不取);
对3g的砝码,给出一个方程 1 + x^3;(1 * x ^ 0 + 1 * x ^ 3,取1个0g的砝码,或是取1个3g的砝码,即3g的砝码取或不取);
(1 + x) * (1 + x ^2) * (1 + x^3) = 1 + x + x^2 + 2 * x^3 + x^4 + x^5 + x^6;
由此我们可以看出,这三个砝码可以称量出1g, 2g, 3g, 4g, 5g, 6g的物体(上式的指数表示可以称量的质量),称3g的物体有两种不同的称法(1,2),(3),称量其他质量的的物体只有一种称法(对应指数表达式前的系数,表示可行的方案数)。
例2:
给定3个数,分别是1,2,3,问:对于一个给定的数n,将n拆分成这3个数,有几种不同的拆法,比如:4 = 1 + 1 + 1 + 1 = 1 + 3 = 2 + 2,有3种不同的拆法;对于这个问题我们可以这样解决:
(1 + x + x^2 + x^3 +...) * (1 + x^2 + x^4 + ...) * (1 + x^3 + x^6 + ...)
展开这个表达式,得到的x^4的系数即为可行的拆分方法数,
看了这两个例子,现在我们比较一下他们二者的异同:
相同之处:都是表达式的指数表示可以的解,对应的系数表示可行的方案数;
不同之处:例1中1g,2g,3g的砝码,每个只可以使用1次;例2中1,2,3这3个数,每个数可以使用无限次;
这道题和例2很类似,对于1个给定的n,我们只要写出这样的式子:(1 + x + x^2 + x^3 +...) * (1 + x^2 + x^4 + ...) * (1 + x^3 + x^6 + ...) * ....*(1 + x^n + x^(2n) + ...),展开这个式子,其中x^n的系数,即为我们所求的方案数;
代码:
#include <cstdio>
const int MAXN = 125;
int c1[MAXN], c2[MAXN];
int main()
{
int n;
while(~scanf("%d", &n))
{
for(int i = 0; i <= n; i++)
{
c1[i] = 1;
c2[i] = 0;
}
for(int i = 2; i <= n; i++)
{
for(int j = 0; j <= n; j++)
{
for(int k = 0; k + j <= n; k += i)
{
c2[k+j] += c1[j];
}
}
for(int j = 0; j <= n; j++)
{
c1[j] = c2[j];
c2[j] = 0;
}
}
printf("%d\n", c1[n]);
}
return 0;
}
代码分析:
数组c1[]用存储最终展开式每项的系数,c1[i]表示x^i的系数,数组c2[]是用来临时存放要计算的下一个表达式的。c1[]表示之前的式子((1 + x + x^2 + x^3 +...) 这样的式子)的相乘之后得到的结果,c2[]表示当前要乘的式子。
for(int i = 0; i <= n; i++)
{
c1[i] = 1;
c2[i] = 0;
}
上面这段代码是初始化,初始化之后,c1[]表示(1 + x + x^2 + x^3 +...);
for(int i = 2; i <= n; i++)
{
for(int j = 0; j <= n; j++)
{
for(int k = 0; k + j <= n; k += i)
{
c2[k+j] += c1[j];
}
}
for(int j = 0; j <= n; j++)
{
c1[j] = c2[j];
c2[j] = 0;
}
}
上面这段代码是用来计算最终的展开式的,思路:得到前面的式子的乘积(保存在c1[]里),再乘上当前式子(c2[]),将得到的结果再放到c1[]里。
第一层循环i(2->n)表示的是当前式子(乘数)为第i个式子,如:i = 2时表示当前式子c2[]为((1 + x^2 + x^4 + ...));
第二层循环j(0->n)表示的是之前的乘积(被乘数)的每一项(x^j这样的项);
第三层循环k(0->n-j)表示的是当前式子(乘数)的每一有意义的项(x^j * x^k = x ^ (j+k), 当 j + k <= n时得到的结果才有意义,因为最后要求的是x^n的系数),相乘得到的结果临时存放在c2[]中,在计算完相乘的结果后,要把c2[]中临时存放的结果放回c1[]中,并且把c2[]清零,为下一轮计算做准备;
最后我们得到的结果就是c1[n]。