有N件物品和一个容量为M的背包。第i件物品的容量是c[i],价值是w[i]。求解将哪些物品装入背包可使价值总和最大。
输入第一行,N,M N件物品和总容量为M,后面N行输入容量和价值,求解背包总价值最大值。
DP主要考虑的是状态转移方程,记DP[i][j]为将第i件物品放入背包中后,背包的总价值,i为第i件物品,j可以理解为背包的剩余容量。
Dp[i][j] = max(dp[i-1][j],dp[i-1][j-c[i]]+w[i]);表示dp[i-1][j]不将第i件物品放入背包,dp[i-1][j-c[i]]+w[i]将第i件物品放入背包。
这里,由于只要返回总价值,多余的状态不用记录,可以只用一个大小为M的一维数组来记录。以下是进行空间压缩的代码。
import java.util.Scanner;
public class Main{
public static void main(String args[]){
Scanner sc = new Scanner(System.in);
int N = sc.nextInt();
int M = sc.nextInt();
int [] need = new int[N];
int [] value = new int[N];
for(int i=0;i<N;i++){
need[i] = sc.nextInt();
value[i] = sc.nextInt();
}
System.out.print(help(need,value,N,M));
}
public static int help(int [] need,int [] value,int N,int M){
int dp [] = new int[M+1];
for(int i=0;i<N;i++){
for(int j=M;j>=need[i];j--){
dp[j] =Math.max(dp[j],dp[j-need[i]]+value[i]);
}
}
return dp[M];
}
}
若需要打印背包里的物品,也就是放的情况是dp[i][j] = dp[i-1][j-c[i]]+w[i]的情况
因此,遍历dp[][]矩阵,满足dp[i][j] = dp[i-1][j-c[i]]+w[i],就打印物品信息c[i]和W[i]
伪代码:
i←N
j←M
while(i>0 && j>0)
do if(F[i][j]=F[i-1][j-C[i]]+W[i])
then Print W[i]
j←j-C[i]
i←i-1
当然也可以定义一个二维数组Path[N][M来存放背包内物品信息,开始时Path[N][M]初始化为0,当 F[i][j]==F[i-1][j-C[i]]+W[i]时Path[i][j]置1。最后通过从Path[N+1][M+1]逆着走向Path[0][0]来获取背包内物品。其中Path[0][]与Path[][0]为边界。伪代码如下:
F[0][] ← {0}
F[][0] ← {0}
Path[][] ← 0
for i←1 to N
do for k←1 to V
F[i][k] ← F[i-1][k]
if(k >= C[i] && F[i][k] < F[i-1][k-C[i]]+W[i])
then F[i][k] ← F[i-1][k-C[i]]+W[i]
Path[i][k] ← 1
return F[N][V] and Path[][]
//打印物品信息
i←N
j←V
while(i>0 && j>0)
do if(Path[i][j] = 1)
then Print W[i]
j←j-C[i]
i←i-1
参考博客:http://blog.youkuaiyun.com/wumuzi520/article/details/7014559