小红书329编程题第二题题解

背包问题。类似的题目有leetcode152, 还有买卖股票的最佳时机III leetcode 123

import java.util.*;
public class Xiaohongshu329_2 {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int x = in.nextInt();
        int[] array = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            array[i] = in.nextInt();
        }
        
        int[][] dp0 = new int[n + 1][x + 1]; //dp0表示没有使用多次推荐的
        int[][] dp1 = new int[n + 1][x + 1]; //dp1表示使用了多次推荐的
        
        for (int i = 1; i <= x; i++) { //初始化
            dp0[0][i] = Integer.MAX_VALUE;
            dp1[0][i] = Integer.MAX_VALUE;
        }
        
        int ans = Integer.MAX_VALUE;
        for (int i = 1; i <= n; i++) {
            int value = array[i];
            int half = array[i] / 2;
            for (int j = 1; j <= x; j++) {
                if (j >= half) { //背包问题
                    dp0[i][j] = Math.min(dp0[i - 1][j], dp0[i - 1][j - half] + 1);
                    dp1[i][j] = Math.min(dp1[i - 1][j], dp1[i - 1][j - half] + 1);
                    if (j >= value) {
                        dp1[i][j] = Math.min(dp1[i][j], dp0[i - 1][j - value] + 1); //在该位置处使用多次推荐
                    }
                } else {
                    dp0[i][j] = dp0[i - 1][j];
                    dp1[i][j] = dp1[i - 1][j];
                }
            }
        }
        ans = Math.min(dp0[n][x], dp1[n][x]);
        if (ans == Integer.MAX_VALUE) {
            System.out.println(-1);
        } else {
            System.out.println(ans);
        }
    }
}

全部评论
m了明天学
1 回复 分享
发布于 03-29 23:28 江苏
如果初始化为Integer.MAX_VALUE,在第一次遍历的时候不会溢出吗
点赞 回复 分享
发布于 03-30 14:12 湖南
佬,题目中要求只有一个号可以多次推荐,您代码中是允许多次进行多次推荐的。比如下例 3 5 1 2 3,是凑不出来的
点赞 回复 分享
发布于 03-30 14:21 江苏

相关推荐

11-15 19:28
已编辑
蚌埠坦克学院 硬件开发
点赞 评论 收藏
分享
评论
12
20
分享
牛客网
牛客企业服务