在谷歌OR工具中保持变量出现的计数

Keeping count of variable occurence in google OR Tools

提问人:MrKhonsu 提问时间:11/11/2023 更新时间:11/11/2023 访问量:28

问:

我正在玩 OR 工具,只是想知道是否有任何方法可以保持满足某个约束的次数?

在这种情况下,我想跟踪我的“diff”变量为 -1、-2、0、2 等的次数。

我已经从官方文档中获得了解决方案打印机,那么我是否也必须以任何方式对其进行编辑/调整以返回计数?

提前致谢,我是新手。

from ortools.sat.python import cp_model
model = cp_model.CpModel()
x = model.NewIntVar(0, 10, 'x')
y = model.NewIntVar(0, 10, 'y')
diff = model.NewIntVar(-10,10,'diff')
model.Add(diff == x - y)
#occurences of each diff

#solution printer
class VarArraySolutionPrinter(cp_model.CpSolverSolutionCallback):
    """Print intermediate solutions."""

    def __init__(self, variables):
        cp_model.CpSolverSolutionCallback.__init__(self)
        self.__variables = variables
        self.__solution_count = 0

    def on_solution_callback(self):
        self.__solution_count += 1
        for v in self.__variables:
            print(f"{v}={self.Value(v)}", end=" ")
        print()

    def solution_count(self):
        return self.__solution_count

solver = cp_model.CpSolver()
# solver.parameters.log_search_progress = True
solution_printer = VarArraySolutionPrinter([x, y, diff])
solver.parameters.enumerate_all_solutions = True
status = solver.Solve(model, solution_printer)
Python 优化 OR-工具 整数编程 运筹学

评论


答:

1赞 Laurent Perron 11/11/2023 #1

为什么不在 on_solution_callback() 方法中增加一个计数器?

评论

0赞 MrKhonsu 11/11/2023
是的,我刚刚想通了。这是非常微不足道的,很抱歉给您带来麻烦。