当前位置:   article > 正文

蓝桥杯DP算法——背包问题(C++)_dp数组背包问题cpp

dp数组背包问题cpp

目录

一、01背包问题

二、完全背包问题

三、多重背包问题

四、多重背包问题(优化版)

五、分组背包问题


一、01背包问题

01背包问题就是有N件物品,一个空间大小为V的背包,每个物品只能使用一次,使得背包中所装物品的价值总和最大。

  如图所示使用一个二维数组来存放从前i个物品中取,总体积不超过j的包中价值最大值。

 根据图二所示,我们可以将每次dp到的情况分为两种,一种是选择第 i 件物品,另一种是不选择第 i 件物品。

  • (不含 i )就是从0~i-1中选择体积不超过 j 的物品,也就是f(i-1,j)
  • (含 i )即从 1 ~ i 中选,包含 i,且总体积不超过 j。可以先把第 i 个物品拿出来,即从第 1 ~ i-1中选,且总体积不超过 j-v[i],也就是f(i-1,j-v_{i})+w_{i}

最终:f[ i ][ j ] = max(f[ i-1 ][ j ], f[i-1][j-v[ i ]] +w[ i ]

例题:https://www.acwing.com/problem/content/2/

朴素做法: 

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

优化: 

 f[i] 只用到了 f[i-1] 这一层,即 f[i-2] 到 f[0] 对 f[i] 是没有用的,所以第二层循环可以直接从 v[i] 开始

  1. for (int i = 1; i <= n; i++) {
  2. for (int j = v[i]; j <= m; j++) {
  3. f[i][j] = max(f[i][j], f[i - 1][j - v[i]] + w[i]);
  4. }
  5. }

 从二维优化成一维(空间优化)

如果直接删除掉 f[i] 这一维即

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

但是这样直接删掉是错误,因为j是递增的。

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

改成一维:f[j] = max(f[j], f[j - v[i]] + w[i]);

由于 f[i][] 只跟上一状态(f[i - 1][])有关 上面两个式子 :这一状态(左) = 上一状态(右)

即 f[i][j] 是由 f[i - 1][j - v[i]] 推出来的 现在进行空间优化,那么必须要保证 f[j] 要由 f[j - v[i]] 推出来的。

如果j层循环是递增的,则相当于 f[i][j] 变得是由 f[i][j - v[i]] 推出来的, 而不是 f[i - 1][j - v[i]] 推出来的。

优化后代码:

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

二、完全背包问题

完全背包不同于01背包的是,每件物品都是无限使用的。

如图所示,与01背包相似的是每次选择0,1,2,3,4,5,....k个

不选时仍是 f[ i-1][ j ]

选择k件时是 f[ i-1][ j-k*v[i]] +k*w[i]

也就是:

                                                         f[ i-1][ j-k*v[i]] +k*w[i]

 例题:https://www.acwing.com/problem/content/3/

朴素写法: 

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

优化:

在对完全背包问题优化时,我们由图公式可知f[i][j]可以简化为max(f[i-1][j],f[i][j-v]+w)

优化后代码:

  1. #include <iostream>
  2. #include <algorithm>
  3. using namespace std;
  4. const int N=1010;
  5. int f[N];
  6. int v[N],w[N];
  7. int n,m;
  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=v[i];j<=m;j++)
  15. {
  16. f[j]=max(f[j],f[j-v[i]]+w[i]);
  17. }
  18. }
  19. cout<<f[m];
  20. return 0;
  21. }

三、多重背包问题

多重背包问题相较于之前的问题就是每个物品是有限s个。 

例题:https://www.acwing.com/problem/content/4/

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

四、多重背包问题(优化版)

首先我们以之前完全背包问题的优化方法尝试使用另一个表达式来代替,但是结果如图所示,不是很好解决。 

二进制法优化:二进制优化的方法在于使用二进制的表示方式来代替每个物品的个数,当某件物品的个数很多的时候无需从1遍历循环到n,可以将其分解成log_{_{2}^{}}^{n}个位权来表示。

为了方便理解举个例子:

如果要拿1001次苹果,传统就是要拿1001次;二进制的思维,就是拿7个箱子就行(分别是装有512、256、128、64、32、8、1个苹果的这7个箱子),这样一来,1001次操作就变成7次操作就行了。

但是要注意的一点是如果某件物品的个数不是二次幂减一,就将前一位的值与s差值记为下一个位权。这样就可以表示0~s之内的所有数了。

然后就将问题分解成为了,01背包问题。

例题:https://www.acwing.com/problem/content/5/

  1. #include<iostream>
  2. using namespace std;
  3. const int N=11010,M=2010;
  4. int v[N],w[N];
  5. int f[N];
  6. int main()
  7. {
  8. int n,m;
  9. cin>>n>>m;
  10. int cnt=0;
  11. while(n--)//初始化v[] w[]
  12. {
  13. int a,b,s;
  14. cin>>a>>b>>s;
  15. int k=1;
  16. while(k<=s)
  17. {
  18. cnt++;
  19. v[cnt]=a*k;
  20. w[cnt]=b*k;
  21. s-=k;
  22. k*=2;
  23. }
  24. if(s>0)
  25. {
  26. cnt++;
  27. v[cnt]=a*s;
  28. w[cnt]=b*s;
  29. }
  30. }
  31. n=cnt;
  32. //01背包
  33. for(int i=1;i<=n;i++)
  34. {
  35. for(int j=m;j>=v[i];j--)
  36. {
  37. f[j]=max(f[j],f[j-v[i]]+w[i]);
  38. }
  39. }
  40. cout<<f[m];
  41. return 0;
  42. }

五、分组背包问题

分组背包问题的不同于之前背包问题的地方在于分组背包是将物品分成几组,然后再在每组里面选择一个物品,并且每组只能选择一个物品。

这里的dp状态计算与之前的也有所不同,这里表示的是第i组选哪一个,f[i-1][j]是表示不选择这一组的物品,f[i-1][j-v[i,k]]+w[i,k]表示的是这一组中选择哪一个。 

例题: https://www.acwing.com/problem/content/9/

  1. #include<iostream>
  2. using namespace std;
  3. const int N=110;
  4. int n,m;
  5. int s[N],v[N][N],w[N][N];
  6. int f[N];
  7. int main()
  8. {
  9. cin>>n>>m;
  10. for(int i=1;i<=n;i++)
  11. {
  12. cin>>s[i];
  13. for(int j=1;j<=s[i];j++)
  14. {
  15. cin>>v[i][j]>>w[i][j];
  16. }
  17. }
  18. for(int i=1;i<=n;i++)
  19. {
  20. for(int j=m;j>=0;j-- )
  21. {
  22. for(int k=0;k<=s[i];k++)
  23. {
  24. if(v[i][k]<=j)
  25. {
  26. f[j]=max(f[j],f[j-v[i][k]]+w[i][k]);
  27. }
  28. }
  29. }
  30. }
  31. cout<<f[m];
  32. return 0;
  33. }

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

闽ICP备14008679号