经典动态规划：0-1 背包问题

天空～之城

于 2023-11-09 18:47:42 发布

阅读量41

点赞数

分类专栏：数据结构与算法文章标签：动态规划算法

数据结构与算法专栏收录该内容

60 篇文章 0 订阅

订阅专栏

转载：https://labuladong.gitee.io/algo/di-er-zhan-a01c6/bei-bao-le-34bd4/jing-dian–28f3c/

在这里插入图片描述

public class Demo01 {
    public static void main(String[] args) {
      int  N = 3, W = 4;
      int[]  wt = {2, 1, 3};
      int[]  val = {4, 2, 3};
      System.out.println(Demo01.knapsack(W,N,wt,val));
    }

    static  int  knapsack(int W, int N, int[] wt, int[] val) {
        assert N == wt.length;
        // base case 已初始化
        int[][] dp = new int[N + 1][W + 1];
        for (int i = 1; i <= N; i++) {
            for (int w = 1; w <= W; w++) {
                if (w - wt[i - 1] < 0) {
                    // 这种情况下只能选择不装入背包
                    dp[i][w] = dp[i - 1][w];
                } else {
                    // 装入或者不装入背包，择优
                    dp[i][w] = Math.max(
                            dp[i - 1][w - wt[i-1]] + val[i-1],
                            dp[i - 1][w]
                    );
                }
            }
        }
        return dp[N][W];
    }
}