HDOJ 1028 Ignatius and the Princess III

链接：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

Problem Description

"Well, it seems the first problem is too easy. I will let you know how foolish you are later." feng5166 says.

"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!"

 

Input

The input contains several test cases. Each test case contains a positive integer N(1<=N<=120) which is mentioned above. The input is terminated by the end of file.

 

Output

For each test case, you have to output a line contains an integer P which indicate the different equations you have found.

 

Sample Input

  
  

   
   4
10
20
  
  

 

Sample Output

  
  

   
   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]。