python-constraint 添加动态约束

Question

我正在使用python约束库来解决CSP的问题，以便为机场的每个航班保留舱位。 我需要将海湾：('A1', 'A2', 'B1', 'B2', 'C1')分配给航班：('MI428', 'UL867', 'QR664', 'TK730', 'UL303')变量集的地方。

将值分配给第二个集合时，几乎没有约束。这是我的代码

from constraint import *
problem = Problem()
flight_names = ['MI428', 'UL867', 'QR664', 'TK730', 'UL303']
bays_list = ['A1', 'A2', 'B1', 'B2', 'C1']
problem.addVariables(flight_names, bays_list)
bay_compat = {'MI428':['A1', 'A2', 'B1'], 'UL867':['B1', 'B2'], 'QR664':['A2', 'B1', 'B2'] , 'TK730':['C1', 'A1'], 'UL303':['B2', 'C1']}
for flight in flight_names:
problem.addConstraint(lambda fl: fl in bay_compat[flight], [flight])
solutions = problem.getSolutions()
print (solutions)

上面的代码工作正常。我想要添加另一个约束，其中每个航班都与特定时间段相关联，称为到达和离开时间之间的时间段。

为此，我创建了另一个列表，如下所示：

time_constraints = {'MI428':(1,3) , 'UL867':(2,7), 'QR664':(3,9), 'TK730':(15,16), 'UL303':(16,17)}

注意：例如，此处 (1,3( 表示凌晨 1：00 至凌晨 3：00

我需要海湾分配，以便同时没有两个航班获得相同的海湾。 所以我问如何在 python-constraint 中使用addConstraint()方法添加该约束？

Answer 1

首先，您必须添加定义航班起飞时间的变量：

flight_names = ['MI428', 'UL867', 'QR664', 'TK730', 'UL303']
flight_times = ['T-' + name for name in flight_names]
bays_list = ['A1', 'A2', 'B1', 'B2', 'C1']
times = range(20)   # or how many times you have
problem.addVariables(flight_names, bays_list)
problem.addVariables(flight_times, times)

然后，您必须施加约束，即每个航班都必须在time_constraints之后分配一定的时间：

time_constraints = {'T-MI428':(1,3) , 'T-UL867':(2,7), 'T-QR664':(3,9), 'T-TK730':(15,16), 'T-UL303':(16,17)}
for flight_time in flight_times:
start, end = time_constraints[flight_time]
problem.addConstraint(lambda fl: fl in range(start, end+1), [flight_time])

现在，您必须添加"您不能在同一海湾同时有两个航班"限制。为此，您需要生成每对可能的航班：

for flight_one, time_one in zip(flight_names, flight_times):
for flight_two, time_two in zip(flight_names, flight_times):
if flight_one == flight_two:
continue
problem.addConstraint(
lambda fl_one, t_one, fl_two, t_two: fl_one != fl_two or t_one != t_two, 
[flight_one, time_one, flight_two, time_two]
)

如果使用的托架不同或使用的时间不同，则满足约束。

---

_{我无法测试上面的代码，但这应该给你一个很好的起点。}