码迷,mamicode.com
首页 > 其他好文 > 详细

完全背包问题

时间:2020-01-14 20:27:46      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:cin   end   cpp   维数   结果   递推   01背包   个数   ++i   

完全背包问题

\(\begin{cases}dp[0][j]=0\\dp[i+1][j]=max(dp[i][j-k*w[i]]+k*v[i]) \end{cases}\)

代码:

int n,W;    cin >> n >> W;
for(int i=0;i<n;++i)
    for(int j=0;j<=W;++j)
        for(int k=0;k*w[i]<=j;++k)
            dp[i+1][j]=max(dp[i+1][j],dp[i][j-k*w[i]]+k*v[i]);
int ans=dp[n][W];

但是这个算法的复杂度达到了\(O(nW^2)\),发现有一不部分计算可以利用前面计算的结果来简化,因此经过精简后的代码:

for(int i=0;i<n;++i)
    for(int j=0; j<=W; ++j)
        if(j<w[i]) dp[i+1][j]=dp[i][j];
        else dp[i+1][j]=max(dp[i][j],dp[i+1][j-w[i]]+v[i]);
ans=dp[n][W];

同时出于节省内存的考虑,可以将其用一维数组表示

for(int i=0;i<n;++i)
    for(int j=w[i];j<=W;++j)
        dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
ans=dp[W];

也放出01背包的代码,对照一下:

for(int i=0;i<n;++i)
    for(int j=W;j>=w[i];--j)
        dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
ans=dp[W];

除此之外,因为递推方程只涉及\(dp[i]\)\(dp[i+1]\)之间的关系,所以可以将两个数组滚动使用

代码:

int dp[2][maxn];
for(int i=0;i<n;++i)
    for(int j=0;j<=W;++j)
        if(j<w[i]) dp[(i+1)&][j]=dp[i&1][j];
        else dp[(i+1)&1][j]=max(dp[i&1][j],dp[(i+1)&1][j-w[i]]+v[i]);

完全背包问题

标签:cin   end   cpp   维数   结果   递推   01背包   个数   ++i   

原文地址:https://www.cnblogs.com/CADCADCAD/p/12193656.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!