牛客 HJ16 购物车问题
最近一段时间在学习算法,在牛客网刷题,遇到一个类似0-1背包的问题,题目链接如下:
牛客HJ16购物车问题
按照自己的思路结合题目描述和0-1背包动态规划法去求解,但结果不对,也不知道什么地方出了问题(自己想的可能价格是10的倍数,初始化模态数组的j列有问题,但想不出来怎么改),或者说自己的思路出现了问题,附自己的代码,求大神指导~
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt(); // 拥有的钱
int m = sc.nextInt(); // 需要购买的数量
int[] money = new int[m+1]; // 存储每个商品的价格
int[] weight = new int[m+1]; //存储每个商品的重要度
int[] flag = new int[m+1]; //存储每个商品是否主件 0-主件 其他-附件&主件编号
// 写入基础数据,规定索引从1开始,方便查找主件
for (int i = 1; i <= m; i++) {
money[i] = sc.nextInt();
weight[i] = sc.nextInt();
flag[i] = sc.nextInt();
}
int maxvalue = Main.method(m, n, money, weight, flag);
System.out.println(maxvalue);
}
public static int method(int m, int n, int[] money, int[] weight, int[] flag){
int value = 0;
int[][] maxtemp = new int[m+1][n/10+1]; // 每个商品价格是10的倍数
//初始化模态数组
for (int i = 0; i <= m; i++) {
maxtemp[i][0] = 0;
}
for (int i = 0; i <= n/10; i++) {
maxtemp[0][i] = 0;
}
int[] isbuy = new int[m+1]; // 标记主件是否已选择 0-未购买 1-已购买
for (int i = 1; i <= m; i++) {
for (int j = 1; j <= n/10; j++) {
if (money[i] > j*10){ // 此商品价格大于拥有的钱
maxtemp[i][j] = maxtemp[i-1][j];
}else{
int buyvalue = 0; // 买当前商品价值
// 判断当前商品是否主件
if (flag[i] == 0){ // 是主件
buyvalue = maxtemp[i-1][j-(money[i]/10)] + (money[i]*weight[i]);
isbuy[i] = 1; //标记主件已选择
}else{
// 判断当前附件所属主件是否选择购买了
// 已选择
if (isbuy[flag[i]] == 1){
buyvalue = maxtemp[i-1][j-(money[i]/10)] + (money[i]*weight[i]);
}
// 未选择 当前价值=购买当前附件价值+所属购买所属主件价值
else{
buyvalue = maxtemp[i-1][j-(money[i]/10)] + (money[i]*weight[i]) + (money[flag[i]]*weight[flag[i]]);
isbuy[flag[i]] = 1;
}
}
// 不买此商品的价值大于买此商品的价值
if ( maxtemp[i-1][j] > buyvalue ) {
maxtemp[i][j] = maxtemp[i-1][j];
}else{
maxtemp[i][j] = buyvalue;
}
}
}
}
value = maxtemp[m][n/10];
return value;
}
}