【蓝桥杯】背包与魔法(java实现)

题目

小蓝面前有 N 件物品, 其中第 i 件重量是 𝑊, 价值是 V 。她还有一个背包, 最大承重是M。

小蓝想知道在背包称重范围内, 她最多能装总价值多少的物品?

特别值得一提的是, 小蓝可以使用一个魔法 (总共使用一次), 将一件物品的重量增加𝐾, 同时价值翻倍。(当然小蓝也可以不使用魔法)
样例输入

3 10 3
5 10
4 9
3 8

输出

26

数据规模
对于 100% 的数据, 1≤𝑁≤2000,1≤𝑀,𝐾≤10000,0≤𝑊𝑖,𝑉𝑖≤10000 ，1≤N≤2000,1≤M,K≤10000,0≤Wi,Vi≤10000.

分析

这道题从题目很容易看出来这是一道动态规划的题目，考察的是经典的o-1背包问题。这道题的附加条件为可以使一件物品的价值翻倍。那么我们就来考虑如何实现这个附加条件。
一开始，我想到的方法是在dp每个重量的同时记录当前重量选择的物品中价值最大的那个，在最后比较实用魔法或者不使用的最大值。（但是该方法有一个测试点过不去，目前我还没有想到是什么原因，大家如果知道原因的话欢迎留言）。
最终，我使用了另一个方法，使用魔法和不使用魔法分开分别dp。

代码

public static void main(String[] args) {
        //读入数据
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int k = sc.nextInt();
        int[] w = new int[n + 1];
        int[] v = new int[n + 1];
        int t = 1;
        while (t <= n) {
            w[t] = sc.nextInt();
            v[t++] = sc.nextInt();
        }
        //创建dp数组，dp[a][0]为重量为a时不使用魔法所得到的价值，dp[a][1]则为使用魔法的价值
        int[][] dp = new int[m + 1][2];
        for (int i = 1; i <= n; i++) {
            //此处从m开始向0遍历，可以省去一个中间数组记录上一次的结果，相当于把上一次的结果直接覆盖了，并且不会使用到本次循环的结果
            for (int j = m; j >= w[i]; j--) {
                dp[j][0] = Math.max(dp[j-w[i]][0] + v[i],dp[j][0]);
                //此处为使用魔法的情况
                if (j-w[i]-k>=0)
                    dp[j][1]=Math.max(dp[j][1],Math.max(dp[j-w[i]][1]+v[i],dp[j-w[i]-k][0]+v[i]*2));
            }
        }
        System.out.println(Math.max(dp[m][0],dp[m][1]));
}