首页 > 代码库 > hdu 1028 Ignatius and the Princess III(母函数,完全背包)

hdu 1028 Ignatius and the Princess III(母函数,完全背包)

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


整数划分问题。

第一道母函数。。。母函数入门


小于等于n的整数共有n个,1,2......n,每个数都有无限多个,对于整数1,它所对应的母函数为(1+x+x^2+...+x^k+...),整数2对应的母函数为(1+x^2+X^4+...+x^(2*k)+...),整数3对应的母函数为(1+x^3+x^6+...+x^(3*k)+...),以此类推,直到整数n。

那么n的整数划分的个数就是这n个母函数乘积(1+x+x^2+...+x^k+...)*(1+x^2+x^4+...+x^(2k)+...)*(1+x^3+x^6+...+x*(3k)+...)+...+(x+x^n+x^(2n)+...)后x^n对应的系数。


#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL long long
#define _LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;

int c1[130],c2[130];

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
    	//初始化,对应第一个表达式(1+x+x^2+...+x^k+...),这时每个i在[0,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++) //j从0到n枚举
            {
                for(int k = 0; k + j <= n; k += i) //k枚举第i个表达式的指数,第i个表达式的增量为i。
                {
                    c2[k+j] += c1[j];
                }
            }
            for(int j = 0; j <= n; j++) //把c2[]的值赋给c1[],因为c2[]每次是从一个表达式开始的。
            {
                c1[j] = c2[j];
                c2[j] = 0;
            }
        }
        printf("%d\n",c1[n]);//x^n的系数就是n的整数划分数。
    }
    return 0;
}



完全背包也可以解。只是把状态转移方程f[i][v] = max( f[i-1][v],f[i-1][v-c[i]]+w[i])的max改成sum。

#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL long long
#define _LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;

int dp[130];

int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		memset(dp,0,sizeof(dp));
		dp[0] = 1;

		for(int i = 1; i <= n; i++)
		{
			for(int j = i; j <= n; j++)
				dp[j] += dp[j-i];
		}

		printf("%d\n",dp[n]);
	}
	return 0;
}