01背包问题

01背包问题

之间觉得这种DP基础题应该掌握的没问题，leetcode上也没找到这个题，后来在Acwing上做了一下，发现一遍过还是有问题，记录一下，留作复习。

题目

有 N 件物品和一个容量是 V 的背包。每件物品只能使用一次。

第 i 件物品的体积是 $v_i$，价值是 $w_i$。

求解将哪些物品装入背包，可使这些物品的总体积不超过背包容量，且总价值最大。
输出最大价值。

输入格式
第一行两个整数，N，V，用空格隔开，分别表示物品数量和背包容积。

接下来有 N 行，每行两个整数 $v_i$,$w_i$，用空格隔开，分别表示第 i 件物品的体积和价值。

输出格式
输出一个整数，表示最大价值。

数据范围
0<N,V≤1000
0<$v_i$,$w_i$≤1000
输入样例
4 5
1 2
2 4
3 4
4 5
输出样例：
8

思路

定义一个二位矩阵f[N][V]，其中f[i][j]表示给定前$i$个物品，背包容量是$j$时，能输出的最大价值，那么状态转移方程为
f [ i ] [ j ] = m a x { f [ i − 1 ] [ j ] , 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[i][j]=max{f[i−1][j],f[i−1][j−wi​]+vi​}，当然，公式还需要满足一定的条件。具体在代码中体现

代码

import java.util.Scanner;

/**
 * Created with IntelliJ IDEA.
 * User: 风云
 * Date: 2020/10/13
 * Time: 16:50
 * Description: 01背包问题
 */
public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int N  =sc.nextInt();
        int V=  sc.nextInt();

        int f[][] = new int[N+1][V+1];
        int W[] = new int[N+1];
        int Value[] = new int[N+1];
        for(int i=0;i<N;i++){
            W[i+1] = sc.nextInt();
            Value[i+1] = sc.nextInt();
        }

        sc.close();
		//前面都是读取数据
        for(int i=1;i<=N;i++){
            for(int j=1;j<=V;j++){
                if(W[i]<=j){
                    f[i][j] = f[i-1][j]>f[i-1][j-W[i]]+Value[i] ? f[i-1][j]:f[i-1][j-W[i]]+Value[i];
                }else{
                    f[i][j] = f[i-1][j];
                }
            }
        }

        System.out.println(f[N][V]);
    }
}

总结

01背包作为DP的基础题，思路还是很简单的，长时间不做还是会忘。
当然，上面的只是基础做法，还可以把二维数据优化成一维，代码如下

import java.util.Scanner;

/**
 * Created with IntelliJ IDEA.
 * User: 风云
 * Date: 2020/10/13
 * Time: 16:50
 * Description: 01背包问题
 */
public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int N  =sc.nextInt();
        int V=  sc.nextInt();

        int f[]= new int[V+1];
        int W[] = new int[N+1];
        int Value[] = new int[N+1];
        for(int i=0;i<N;i++){
            W[i+1] = sc.nextInt();
            Value[i+1] = sc.nextInt();
        }

        sc.close();

        for(int i=1;i<=N;i++){
            for(int j=V;j>=0;j--){//反转一下遍历顺序即可
                if(W[i]<=j){
                    f[j] = f[j]>f[j-W[i]]+Value[i] ? f[j]:f[j-W[i]]+Value[i];
                }
            }
        }

        System.out.println(f[V]);
    }
}