洛谷 P1048 采药 01背包模板题

洛谷 P1048 采药
01背包问题,每个物品只能拿一次,且与拿物品的顺序无关,相当于组合。
不想要第 i 棵树,和放不进去是一样的,价值和空间都没有变化,也就是前 i-1 棵树 j 时间的最大价值。
如果要第 i 棵树,前 i - 1 棵树 j - arr[i - 1][0] 时间的最大价值加上第 i 棵树的价值,因为需要给第 i 棵树留出空间,所以前面 i -1棵树就只剩下 j - arr[i -1][0]的空间了,所以这时候的价值就是 i - 1 棵树 j - arr[i - 1][0]时间加上第 i棵树的价值
因为放进第 i 棵树会对前 i - 1棵树的利益造成冲击,所以需要取放与不放的两种选择的更大的价值

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int t = in.nextInt(); // 总时间
        int m = in.nextInt(); // 共有m株草药
        int[][] arr = new int[m][2];
        for (int i = 0; i < m; i++) {
            arr[i][0] = in.nextInt(); // 时间
            arr[i][1] = in.nextInt(); // 价值
        }
        // 前i颗树花费j时间的最大价值
        int[][] value = new int[m+1][t+1];
        // 赋初始值为,第一列不管多少个物品空间为0,所以价值为0
        // 第一行0个物品,所以价值为0
        
        for (int i = 1; i <= m; i++) {
            for (int j = 1; j <= t; j++) {
                // 如果第i棵树的时间大于时间j,放不进去,
                // 只能等于前i-1棵树j时间的最大价值,这个时候空间和价值都没有变化,所以等于value[i-1][j]
                if (j < arr[i-1][0]) {
                    value[i][j] = value[i-1][j];
                }
                else {
                    value[i][j] = Math.max(value[i-1][j], value[i-1][j-arr[i-1][0]] + arr[i-1][1]);
                }
            }
        }
        System.out.println(value[m][t]);
    }
}

空间还可以压缩成一个一维数组,因为算第 i 棵树的时候只与第 i - 1 行算出的状态有关,所以只需要一个长度为 t(时间) 的数组来记录状态

class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int t = in.nextInt(); // 总时间
        int m = in.nextInt(); // 共有m株草药
        int[][] arr = new int[m][2];
        for (int i = 0; i < m; i++) {
            arr[i][0] = in.nextInt(); // 时间
            arr[i][1] = in.nextInt(); // 价值
        }
        // 空间压缩成一个一维数组
        int[] value = new int[t+1];

        for (int i = 1; i <= m; i++) {
            // 注意这里是从后面向前遍历,因为第 i 棵树用到的是第 i-1 棵树的状态
            // 如果从前往后遍历的话第 i 棵树的状态就会覆盖掉第 i-1 棵树的状态
            for (int j = t; j >= 1; j--) {
                // value[j] 表示不拿第 i 棵树,也就是二维数组表示的 value[i-1][j]
                // value[j-arr[i-1][0]] + arr[i-1][1] 表示拿第 i 棵树,也就是二维数组的 value[i-1][j-arr[i-1][0]] + arr[i-1][1]
                if (j-arr[i-1][0] >= 0) {
                    value[j] = Math.max(value[j], value[j-arr[i-1][0]] + arr[i-1][1]);
                }
            }
        }
        System.out.println(value[t]);
    }
}

还有一个初始化的问题,采药这道题问的是总价值最大,并没有要求要让背包装满,所以初始值可以初始化为0,(因为什么都不放就是一种放的方法,价值为0),而有的题目要求一定要恰好让背包装满,则初始值需要初始化为无穷大(小),表示不能装满。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值