记录一下在AcWing中的学到的完全背包问题,同时希望在写给过程中对这个问题有一个更好的理解。
朴素版
import java.util.*;
public class Main{
static int N = 1010;
static int[] v = new int[N];
static int[] w = new int[N];
static int n,m;
static int[][] f = new int[N][N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
for(int i = 1; i <= n; i++){
v[i] = sc.nextInt();
w[i] = sc.nextInt();
}
for(int i = 1; i <= n; i++){
for(int j = 0; j <= m; j++){
for(int k = 0; k * v[i] <= j; k++){
f[i][j] = Math.max(f[i][j],f[i-1][j-v[i]*k]+k*w[i]);
}
}
}
System.out.print(f[n][m]);
}
}
作者:QSS
链接:https://www.acwing.com/activity/content/code/content/6516012/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
优化成二维
import java.util.*;
public class Main{
static int N = 1010;
static int[] v = new int[N];
static int[] w = new int[N];
static int n,m;
static int[][] f = new int[N][N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
for(int i = 1; i <= n; i++){
v[i] = sc.nextInt();
w[i] = sc.nextInt();
}
for(int i = 1; i <= n; i++){
for(int j = 0; j <= m; j++){
f[i][j] = f[i-1][j];
if(j >= v[i]){
f[i][j] = Math.max(f[i][j],f[i][j-v[i]]+w[i]);
}
}
}
System.out.print(f[n][m]);
}
}
作者:QSS
链接:https://www.acwing.com/activity/content/code/content/6516012/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这里是将原来的三维优化成了二维,我就来讲述一下怎么优化的吧(其实还是根据y总的教学视频来的)。
原来的三维的转换方程是:
我们要减掉的是k的那一维(我觉得先这么做的原因是因为先将k这一维去掉后就比较向01背包的那种样式,其实先去掉i的那一维应该也是一样的,看个人观点吧,但是这里根据视频来讲的话就先去掉k的这一维)。
分析结果如下图:
优化成一维
import java.util.*;
public class Main{
static int N = 1010;
static int[] v = new int[N];
static int[] w = new int[N];
static int n,m;
static int[] f = new int[N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
for(int i = 1; i <= n; i++){
v[i] = sc.nextInt();
w[i] = sc.nextInt();
}
for(int i = 1; i <= n; i++){
for(int j = v[i]; j <= m; j++){
f[j] = Math.max(f[j],f[j-v[i]]+w[i]);
}
}
System.out.print(f[m]);
}
}
作者:QSS
链接:https://www.acwing.com/activity/content/code/content/6516012/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
在这里有二维转换成一维的话其实跟01背包有点像,但是有点不同,就是在01背包时要将j的顺序变成递减的顺序,但是在这里不需要,01背包那里变是因为他需要的i-1层的值,所以要变,但是这里就是需要的第i层的,所以j的顺序就是递增的就可以了。
关于01背包的优化问题可以看同专栏AcWing里的AcWing2-01背包(朴素版+优化版)。
如果有问题,希望各位网友能够指出,我先提前谢谢各位了。