У меня есть набор из многих (более 10000) предметов, из которых я должен выбрать ровно 20 предметов. Я могу выбрать каждый товар только один раз. Мои вещи имеют прибыль и затраты, а также несколько логических свойств (таких как цвет).
Я прочитал и работал над учебником в https://developers.google.com/optimization/mip/integer_opt_cp и https://developers.google.com/optimization/mip/integer_opt,, но мои ограничения немного отличаются от тех, которые там представлены.
Каждый элемент представлен в виде кортежа:
item = ('item name', cost, profit, is_blue)
в качестве примера
vase = ['Ming Vase', 1000, 10000, 0]
plate = ['China Plate', 10, 5, 1]
, а общий набор предметов представляет собой список списков:
items = [item1, item2, ..., itemN].
Мои доходы и расходы также перечислены:
profits = [x[2] for x in items]
costs = [x[1] for x in items]
Для каждого выбранного элемента он должен иметь минимальное значение, а минимум 5 элементов должны иметь флаг свойства (is_blue), равный 1.
Я хочу выбрать 20 самых дешевых предметов с наибольшим значением, чтобы у 5 из них флаг свойства был установлен на 1.
У меня проблемы с формулировкой с помощью инструментов Google OR.
from ortools.linear_solver import pywraplp
solver = pywraplp.Solver('SolveAssignmentProblemMIP',
pywraplp.Solver.CBC_MIXED_INTEGER_PROGRAMMING)
x = {}
for i in range(MAX_ITEMS):
x[i] = solver.BoolVar('x[%s]' % (i))
#Define the constraints
total_chosen = 20
solver.Add(solver.Sum([x[i] for i in range(MAX_ITEMS)]) == total_chosen)
max_cost = 5.0
for i in range(num_recipes):
solver.Add(x[i] * cost[i] <= max_cost)
solver.Maximize(solver.Sum([profits[i] * x[i] for i in range(total_chosen)]))
sol = solver.Solve()
Я могу получить набор предметов, которые я выбрал:
for i in range(MAX_ITEMS):
if x[i].solution_value() > 0:
print(item[i].item_name)
Это прекрасно работает - он выбирает набор из 20 элементов, которые максимизируют прибыль в зависимости от ограничения затрат, но я застрял, как расширить это, чтобы выбрать элементы с свойством (is_blue), установленным в true или false.
Любая помощь в формулировании ограничений и целей была бы действительно полезной. Спасибо!