我正在从here修改代码，我希望能够为某一天添加不同长度的轮班(例如，我希望星期五/第4天只有2个轮班)。我的代码总是以错误代码结尾。我相信这是由于我设定的一些限制的内部问题。在
我在类似的程序上看到了一些关于StackOverflow的帖子，但是找不到一个能解决我的具体问题的帖子。在from ortools.sat.python import cp_model
class employeePartialSolutionPrinter(cp_model.CpSolverSolutionCallback):
"""Print intermediate solutions."""
def __init__(self, shifts, num_employee, num_days, num_shifts, sols):
cp_model.CpSolverSolutionCallback.__init__(self)
self._shifts = shifts
self._num_employee = num_employee
self._num_days = num_days
self._num_shifts = num_shifts
self._solutions = set(sols)
self._solution_count = 0
def on_solution_callback(self):
self._solution_count += 1
if self._solution_count in self._solutions:
print('Solution %i' % self._solution_count)
for d in range(self._num_days):
print('Day %i' % d)
for n in range(self._num_employee):
is_working = False
for s in range(self._num_shifts):
if self.Value(self._shifts[(n, d, s)]):
is_working = True
print(' Employee %i works shift %i' % (n, s))
if not is_working:
print(' Employee {} does not work'.format(n))
def solution_count(self):
return self._solution_count
model = cp_model.CpModel()
solver = cp_model.CpSolver()
num_employee = 5
num_shifts = 5
num_days = 5
all_employee = range(num_employee)
all_shifts = range(num_shifts)
all_days = range(num_days)
# Normal Hours
# Monday-Thursday Shift 0-4
# Friday Shift 0-1
friday_deduct = 3
shifts = {}
for n in all_employee:
for d in all_days:
if d == 4:
for s in range(num_shifts-friday_deduct):
shifts[(n, d, s)] = model.NewBoolVar('shift_n%id%is%i' % (n, d, s))
else:
for s in all_shifts:
shifts[(n,d,s)] = model.NewBoolVar('shift_n%id%is%i' % (n,d,s))
"""
Constraints (Normal Time)
"""
# Each Shift is assigned to a single person per day
# Shift 2 need to be assigned to 3 person
# Shift 1 and 3 need to be assigned to 2 person
for d in all_days:
if d == 4:
for s in range(num_shifts-friday_deduct):
if s == 1:
model.Add(sum(shifts[(n, d, s)] for n in all_employee) == 2)
else:
model.Add(sum(shifts[(n, d, s)] for n in all_employee) == 1)
else:
for s in all_shifts:
if s == 2 :
model.Add(sum(shifts[(n,d,s)] for n in all_employee) == 3)
elif s == 3 or s == 1:
model.Add(sum(shifts[(n, d, s)] for n in all_employee) == 2)
else:
model.Add(sum(shifts[(n,d,s)] for n in all_employee) == 1)
#Each nurse works at most 10 shift per week, at least 4 shift per week
for n in range(num_employee):
week = []
for d in all_days:
if d == 4:
for s in range(num_shifts-friday_deduct):
week.append(shifts[(n,d,s)])
# week.append(sum(shifts[(n, d, s)] for s in range(num_shifts-friday_deduct)))
else:
for s in all_shifts:
week.append(shifts[(n,d,s)])
# week.append(sum(shifts[(n,d,s)] for s in all_shifts))
model.Add(sum(week) >= 4)
model.Add(sum(week) <=10)
solver.parameters.linearization_level = 0
a_few_solutions = range(5)
solution_printer = employeePartialSolutionPrinter(shifts, num_employee,
num_days, num_shifts, a_few_solutions)
solver.SearchForAllSolutions(model, solution_printer)
这是Pycharm IDE的打印输出。当我从命令行运行它时，会出现窗口“Python has stop working”(Python已停止工作)。在
^{pr2}$
)
