当前位置:   article > 正文

c++01背包问题_c++ 01背包

c++ 01背包

01背包问题是一个经典的动态规划问题,它的基本形式是:有一个背包,它的容量为C(Capacity)。现在有n个物品,每个物品的重量为w[i],价值为v[i]。问你如何选择装入背包的物品,使得装入背包中的物品的总价值最大?

其中,每个物品只有一个,要么装入背包,要么不装,因此称为01背包问题。

以下是C++实现01背包问题的代码:

  1. #include <iostream>
  2. #include <algorithm>
  3. #include <cstring>
  4. using namespace std;
  5. const int N = 1010;
  6. int n, m;
  7. int w[N], v[N];
  8. int f[N][N];
  9. int main()
  10. {
  11. cin >> n >> m;
  12. for(int i = 1; i <= n; i++)
  13. cin >> w[i] >> v[i];
  14. for(int i = 1;i <= n;i++)
  15. for(int j = 0;j <= m;j++)
  16. {
  17. f[i][j] = f[i-1][j]; //不选当前物品
  18. if(j >= w[i]) //剩余空间大于当前物品的大小
  19. f[i][j] = max(f[i][j], f[i-1][j-w[i]]+v[i]); //选当前物品
  20. }
  21. cout<<f[n][m]<<endl;
  22. return 0;
  23. }

在该代码中,我们使用f[i][j]表示选前i个物品,体积为j时的最大价值,其中i=1,2,3,...,n,j=0,1,2,...,m。

由于每个物品只能选一次,因此我们可以将问题划分为两种情况:选或不选当前物品。

如果不选当前物品,则最大价值就是前i-1个物品中,体积为j的最大价值f[i-1][j]。

如果选当前物品,则最大价值就是前i-1个物品中,体积为j-w[i]的最大价值加上当前物品的价值v[i],即f[i-1][j-w[i]]+v[i]。

综上所述,我们可以得到状态转移方程:

f[i][j] = max(f[i-1][j], f[i-1][j-w[i]]+v[i]);

最后,我们输出f[n][m]即可得到最大价值。

注:该代码时间复杂度为O(nm),可以通过优化空间复杂度将其优化为O(m)。

现在我们就来优化一下

老规矩,先给代码

  1. #include<iostream>
  2. #include<algorithm>
  3. using namespace std;
  4. const int N=1010;
  5. int n,m;
  6. int v[N],w[N];
  7. int f[N];
  8. int main()
  9. {
  10. cin>>n>>m;
  11. for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
  12. for(int i=1;i<=n;i++)
  13. {
  14. for(int j=m;j>=v[i];j--)
  15. {
  16. f[j]=max(f[j],f[j-v[i]]+w[i]);
  17. }
  18. }
  19. cout<<f[m]<<endl;
  20. return 0;
  21. }

区别:j从m开始

从大到小开始,因为我们在用到第i行的时候,实际上我们用的是第i-1行,这就说明所有我们不需要保存所有的数据,只需要上一层的数据。可以理解dp[][]仍为2维,但是第一维长度为1.只有一层。保存上一层。
逆序是为了保存更新本层时,使用的是上一层的原始数据。

所以我们可以在体积最大的时候枚举,以保证取得最大值

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/432290
推荐阅读
相关标签
  

闽ICP备14008679号