我有一个类似于0-1 Knapsack的问题。
我试图做的修改是获取选定对象的列表,而不是成本。我尝试使用以下代码:
public static int fillPackage(double weight,ArrayList<Item> item, int n) {
//base case
if(n == 0 || weight == 0)
return 0;
if(item.get(n - 1).getWeight() > weight)
return fillPackage(weight, item, n - 1);
else {
int include_cost = item.get(n - 1).getCost() + fillPackage((weight - item.get(n - 1).getWeight()), item, n - 1);
int exclude_cost = fillPackage(weight, item, n - 1);
if(include_cost > exclude_cost) {
my_pack.add(item.get(n - 1));
return include_cost;
}
else {
return exclude_cost;
}
}
}
这里
my_pack
是用来存储选定对象的ArrayList
。但是,它也在存储其他对象另外,我不能使用dp table方法,因为我的参数是
float
。应该在哪里放置
my_pack.add()
行? 最佳答案
问:
应该在哪里放置my_pack.add()
行?
答:
把my_pack.add()
放在任何地方都不能得到正确的代码。
我来告诉你怎么做。你知道解决0-1 Knapsack
的方法是:
best_cost = max(included_cost, excluded_cost)
同样地,你应该这样思考你的问题:
if included_cost > excluded_cost
best_choice = included_best_choice + included_item
else
best_choice = excluded_best_choice
我修改了您的代码,它可以正确地解决您的问题(您可以编辑和运行我的测试代码)。请看下面的代码。
如果您有任何问题,请留言,我会尽快回复。
import java.util.ArrayList;
import java.util.List;
public class Package {
static List<Item> my_pack;
public static int fillPackage(double weight, ArrayList<Item> item, List<Item> optimalChoice, int n){
//base case
if(n == 0 || weight == 0)
return 0;
if(item.get(n-1).getWeight() > weight) {
List<Item> subOptimalChoice = new ArrayList<>();
int optimalCost =fillPackage(weight, item, subOptimalChoice, n-1);
optimalChoice.addAll(subOptimalChoice);
return optimalCost;
}
else{
List<Item> includeOptimalChoice = new ArrayList<>();
List<Item> excludeOptimalChoice = new ArrayList<>();
int include_cost = item.get(n-1).getCost() + fillPackage((weight-item.get(n-1).getWeight()), item, includeOptimalChoice, n-1);
int exclude_cost = fillPackage(weight, item, excludeOptimalChoice, n-1);
if(include_cost > exclude_cost){
optimalChoice.addAll(includeOptimalChoice);
optimalChoice.add(item.get(n - 1));
return include_cost;
}
else{
optimalChoice.addAll(excludeOptimalChoice);
return exclude_cost;
}
}
}
public static void main(String args[]) {
ArrayList<Item> itemList = new ArrayList<>();
itemList.add(new Item(2, 1));
itemList.add(new Item(5, 6));
itemList.add(new Item(3, 2));
itemList.add(new Item(4, 4));
itemList.add(new Item(7, 7));
printOptimalChoice(itemList, 9);
printOptimalChoice(itemList, 10);
printOptimalChoice(itemList, 11);
}
private static void printOptimalChoice(ArrayList<Item> itemList, double weight) {
my_pack = new ArrayList<>();
fillPackage(weight, itemList, my_pack, itemList.size());
System.out.println("Best choice for weight: " + weight);
for(int i = 0; i < my_pack.size(); i++) {
System.out.println(my_pack.get(i));
}
}
}
我的测试输出:
Best choice for weight: 9.0
Item{weight=5.0, cost=6}
Item{weight=4.0, cost=4}
Best choice for weight: 10.0
Item{weight=5.0, cost=6}
Item{weight=4.0, cost=4}
Best choice for weight: 11.0
Item{weight=2.0, cost=1}
Item{weight=5.0, cost=6}
Item{weight=4.0, cost=4}
代码:
class Item {
private double weight;
private int cost;
public Item(double weight, int cost) {
this.weight = weight;
this.cost = cost;
}
@Override
public String toString() {
return "Item{" +
"weight=" + weight +
", cost=" + cost +
'}';
}
public double getWeight() {
return weight;
}
public void setWeight(double weight) {
this.weight = weight;
}
public int getCost() {
return cost;
}
public void setCost(int cost) {
this.cost = cost;
}
}