【01背包问题】
有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。
第 i 件物品的体积是 vi,价值是 wi。
求解将哪些物品装入背包,可使这些物品的总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行两个整数,N,V,用空格隔开,分别表示物品数量和背包容积。
接下来有 N 行,每行两个整数 vi,wi,,用空格隔开,分别表示第 ii件物品的体积和价值。
输出格式
输出一个整数,表示最大价值。
数据范围
0<N,V≤10000<N,V≤1000
0<vi,wi≤1000
代码(二维数组版):
private static void Bag_01(){
int N=1010;
Scanner in =new Scanner(System.in);
int n,m;//n物品个数,m背包体积
n=in.nextInt();
m=in.nextInt();
int[] v=new int[n+1];//物品数组
int[] w=new int[n+1];//物品价值数组
int[][] dp=new int[N][N];//总价值
for(int i=1;i<=n;i++){
v[i]=in.nextInt();
w[i]=in.nextInt();
}
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
dp[i][j]=dp[i-1][j];
if(j>=v[i]){
dp[i][j]=Math.max(dp[i][j],dp[i-1][j-v[i]]+w[i]);
}
}
System.out.println("\n");
System.out.println("第"+i+"个物品");
for(int k=0;k<n;k++){
for(int t=0;t<=m;t++){
System.out.print(dp[k][t]+" ");
}
System.out.println("\n");
}
}
System.out.println(dp[n][m]);
}
dp[i][j]=Math.max(dp[i][j],dp[i-1][j-v[i]]+w[i]);
分析:
在第i个物品时
要么不放入,维持前i-1个物品的价值
要么放入,dp[i-1][j-v[i]]+w[i],就是前i-1个物品的最大价值+当前要选的第i物品的价值,但是我一直理解不了为什么是遍历j从0~最带,为什么要写成j-v[i]。所以自己总结了个理解。配上二维数组图会更加了然,视频:https://www.bilibili.com/video/BV1XD4y1S7nv?from=search&seid=6022281859510960430
我理解是:dp[i-1][j-v[i]]+w[i]为直接把前i-1个物品先拿出来,再放入第i个物品(有if判断,所以拿出前i-1个物品后一定放得下),放入后,dp加上第i个物品的价值,然后算剩余体积=当前背包体积(j)-第i个物品的体积(v[i]),根据这个剩余体积的大小往前(二维数组)去找前i-1个物品中的最大价值。
代码(一维数组空间优化版):
private static void Bag_01you(){
int N=1010;
Scanner in =new Scanner(System.in);
int n,m;//n物品个数,m背包体积
n=in.nextInt();
m=in.nextInt();
int[] v=new int[n+1];//物品数组
int[] w=new int[n+1];//物品价值数组
int[] dp=new int[N];//总价值
for(int i=1;i<=n;i++){
v[i]=in.nextInt();
w[i]=in.nextInt();
}
for(int i=1;i<=n;i++){
// for(int j=0;j<=m;j++){
// if(j>=v[i]){
// dp[j]=Math.max(dp[j],dp[j-v[i]]+w[i]);
// }
// }
for(int j=m;j>=v[i];j--){
if(j>=v[i]){
dp[j]=Math.max(dp[j],dp[j-v[i]]+w[i]);
}
}
System.out.println("第"+i+"个物品");
for(int k=0;k<6;k++){
System.out.print(dp[k]+" ");
}
System.out.println("\n");
}
System.out.println(dp[m]);
}
这里要注意遍历背包大小j是逆序,用了好久才理解
我们把dp[i-1][j-v[i]]改成了dp[j-v[i]],如果是顺序,dp[j-v[i]]并非是前i-1个物品的最大价值,而是当前第i个物品,这就会造成重复的问题。当然,这一句话我自己一开始都不太理解,所以用一个过程来说明
我们假设输入是:
4 5
1 2
2 4
3 4
4 5
如果j是正序遍历
for(int j=0;j<=m;j++){
if(j>=v[i]){
dp[j]=Math.max(dp[j],dp[j-v[i]]+w[i]);
}
}
i=1时
dp[0]=0 dp[1]=2 dp[2]=max(dp[2],dp[2-1]+2)=4
可以看到dp[2]的时候,用到的dp[2-1]即dp[1]是当前i=1的dp[1]而不是i-1的dp[1],dp[2]出错,后面都是错的
如果j是逆序遍历
for(int j=m;j>=v[i];j--){
if(j>=v[i]){
dp[j]=Math.max(dp[j],dp[j-v[i]]+w[i]);
}
}
j递减,j-v[i]递减,j-v[i]一定是当前第i个物品没有算过的,而是前i-1的值
i=1时
dp[5]=max(dp[5],dp[5-1]+2)=2
dp[4]=max(dp[4],dp[4-1]+2)=2
dp[3]=max(dp[3],dp[3-1]+2)=2
dp[2]=max(dp[2],dp[2-1]+2)=2
dp[1]=max(dp[1],dp[1-1]+2)=2
i=2时
dp[5]=max(dp[5],dp[5-2]+4)=max(dp[5],dp[3]+4)=6
dp[4]=max(dp[4],dp[4-2]+4)=max(dp[4],dp[2]+4)=6
dp[3]=max(dp[3],dp[3-2]+4)=max(dp[3],dp[1]+4)=6
dp[2]=max(dp[2],dp[2-2]+4)=max(dp[2],dp[0]+4)=4