DP-母函数

来源:互联网 发布:阿里云服务器流量攻击 编辑:程序博客网 时间:2024/04/30 15:26

先由钱币兑换问题开始
http://acm.hdu.edu.cn/showproblem.php?pid=1284 
Problem Description
在一个国家仅有1分,2分,3分硬币,将钱N兑换成硬币有很多种兑法。请你编程序计算出共有多少种兑法。
Input
每行只有一个正整数N,N小于32768。
Output
对应每个输入,输出兑换方法数。


这道题有三种解法(参照此博客http://www.cnblogs.com/Findxiaoxun/p/3574907.html)
完全背包的解法很容易想到,模板性质的。
第二种技巧性的就会强一点。这样想,先只考虑1和3,不是1,就是3,全1的只有一种;每三个1就可以兑换一个3,方法数就有n/3种。再考虑2的情况,全部是2和1的情况,实际上,去掉i个3,就可以变成只含有1和2的情况,而类似的,只含有1和2的情况,可以有(n-3*i)/2种,此时只需要把这些方法数加起来就可以了。
第三种就是母函数了,在此引用下这位大神的博客(http://www.wutianqi.com/?p=596),Woo讲的很好了,尤其是算法归纳的过程,不过个人感觉少了对代码的分析,这里我毛遂自荐,来讲下我的理解:
大家在学习母函数的时候,一定要记住理解这样一句话:
普通母函数代码实现其实就是模拟手算多项式乘法。
首先要看Woo的那篇文章,最重要的理解是:


①  、首先对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每次是从一个表达式中开始的。


Woo这里是根据题目来说的,而且,其实他的代码里Num那个地方,其实是有两种的,一般情况下。
来看下母式子:
(1+x+x^2+x^3+x^4+..)(1+x^2+x^4++..)(1+x^3+x^6+..)
第一个括号指的是1分的,在拆分第一个括号的之前,1分的能够构成数m,那么c1[m]=1;如果是有a个1分的,那么1分一直a,c1[a]=1;然后开始拆分第一个括号,就是把第一个括号和第二个括号合并,我们手算多项式乘法,先按括号一的那个1,乘(第二个括号里的内容),很自然的c2[i]+=c1[i],这里的c2[i]表示的是在这次的循环过程中,中间的结果,这句话理解不了的话,继续看。然后括号一的第二个元素x,x*(....)=x*1+x*x^2+x*x^4...=x+x^3+x^5...最终,它们会和第一次乘出来的结果同项合并,x^3+x^3=2*x^3,其他值都类似,那么,c2[3]+=c1[1];c1[i]就是保存的之前乘出来的x^i结果的系数,等这一次乘完,c2[i]保存了最终的结果,那就把它的值都转移到c1里面去,然后c2又空。继续分解括号。
举个例子:
3个1分的硬币,1个2分的,2个3分的。
(1+x+x^2+x^3)(1+x^2)(1+x^3+x^6)

初始化:

      0 1 2 3 4 5 6 7 8 9

c1: 1 1 1 1 0 0 0 0 0 0

C2全0

对第一个括号的1*(1+x^2),(i=2,j=0)得到

        0 1 2 3 4 5 6 7 8 9

C2: 1 0 1 0 0 0 0 0 0 0


就是说,原来2个1可以组成2,现在,一个2就能替换,有1种方法。

X*(1+x^2) (i=2;j=1)

        0 1 2 3 4 5 6 7 8 9

C2: 1 1 1 1 0 0 0 0 0 0


X^2*(1+x^2)  j=2

   0 1 2 3 4 5 6 7 8 9

C2: 1 1 2 1 0 0 0 0 0 0

x^3*(1+x^2)   j=3

  0 1 2 3 4 5 6 7 8 9

C2:1 1 2 2 0 0 0 0 0 0


把c2的值全都转移到c1里
i=3
剩下的步骤也是如此。
就是模拟手工计算多项式乘法。


Woo的博客介绍的几道题都挺不错,现在外面来把这个算法真正的理解运用下:HDU1171,题意就是说,给价值不同的一些物品,让你把它们分成和尽可能接近的两堆。一种很巧妙的思路是,转换成完全背包,或者是01背包。因为动态规划解决的问题一般是最××的问题,最多,最少,最接近,等等。而这个题,可以看成是sum/2空间的背包,让你用那些物品装,尽可能装满。背包的代码我先附上,背包专辑我后续我会整理出来。
#include<cstdio>#include<algorithm>#include<cstring>using namespace std;const int maxn=5005;const int maxm=255555;int n,t,sum;int dp[maxm],v[maxn];int main(){    while(scanf("%d",&t)&&t>0){        int a,b;        n=0;sum=0;        memset(v,0,sizeof(v));        for(int i=0;i<t;i++){            scanf("%d%d",&a,&b);            while(b--){                v[n++]=a;//wa                sum+=a;            }        }        int sum2=sum/2;        memset(dp,0,sizeof(dp));        for(int i=0;i<n;i++){            for(int j=sum2;j>=v[i];j--){                dp[j]=max(dp[j],dp[j-v[i]]+v[i]);            }        }//        for(int i=0;i<=sum2;i++)printf("%d  ",dp[i]);        int ans=max(dp[sum2],sum-dp[sum2]);        printf("%d %d\n",ans,sum-ans);    }    return 0;}



再来看母函数的思路。可以这样想,因为题中对价值的限制为50,也就是说,只有50种值,有50种面值个数不同的硬币,要你求出,能表示出的最接近总数一半的那个值。转换为了母函数问题。来看i,j,k
50种面值,1的可以直接先处理,那么i=2 to 50,而j的范围就是0 to sum,k则是(k=0;k+j<=sum&& k<i*num[i];k+=i)k的限制也很好理解,因为只有num[i]个i面值的硬币,则多项式最多就能乘到x^(i*num[i])。

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int c1[250005],c2[250005];int data[51];int sum;int main(){    int n;    while(scanf("%d",&n)&&n>0){        sum=0;        int a,b;        memset(data,0,sizeof(data));        memset(c1,0,sizeof(c1));        memset(c2,0,sizeof(c2));        for(int i=0;i<n;i++){            scanf("%d%d",&a,&b);            data[a]=b;            sum+=a*b;        }        for(int i=0;i<=data[1];i++)c1[i]=1;        for(int i=2;i<=50;i++){            for(int j=0;j<=sum;j++){                for(int k=0;j+k<=sum&&k<=i*data[i];k+=i)                    c2[j+k]+=c1[j];            }            for(int j=0;j<=sum;j++){                c1[j]=c2[j];c2[j]=0;            }        }        int x=sum/2;        while(!c1[x]){x--;}        int ans=max(x,sum-x);        printf("%d %d\n",ans,sum-ans);    }    return 0;}



0 0
原创粉丝点击