题解 | #完全背包#

完全背包

http://www.nowcoder.com/practice/3ed13831e2cc4613866edee237d5a804

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param v int整型 
     * @param n int整型 
     * @param nums int整型ArrayList<ArrayList<>> 
     * @return int整型ArrayList
     */
    public ArrayList<Integer> knapsack(int v, int n, ArrayList<ArrayList<Integer>> nums) {
        // write code here
        int[] V = new int[n + 1];
        int[] W = new int[n + 1];
        for (int goods = 1; goods <= n; goods++) {
            V[goods] = nums.get(goods - 1).get(0);
            W[goods] = nums.get(goods - 1).get(1);
        }
        int[] dp1 = new int[v + 1];
        int[] dp2 = new int[v + 1];
        Arrays.fill(dp2, Integer.MIN_VALUE);
        dp2[0] = 0;
        for (int goods = 1; goods <= n; goods++) {
            for (int capacity = V[goods]; capacity <= v; capacity++) {
                dp1[capacity] = Math.max(dp1[capacity], dp1[capacity - V[goods]] + W[goods]);
                dp2[capacity] = Math.max(dp2[capacity], dp2[capacity - V[goods]] + W[goods]);
                if (dp2[capacity] < 0) {
                    dp2[capacity] = Integer.MIN_VALUE;
                }
            }
        }
        ArrayList<Integer> res = new ArrayList<>();
        res.add(dp1[v]);
        res.add(dp2[v] == Integer.MIN_VALUE ? 0 : dp2[v]);
        return res;
    }
}
全部评论

相关推荐

10-09 19:35
门头沟学院 Java
洛必不可达:java的竞争激烈程度是其他任何岗位的10到20倍
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务