我认为有一个隐含的假设,即所有三种颜色的花朵都出现在花园中。考虑到这一点,这就是我将如何使用 Python 和 Haskell 接口对 Z3 进行编码的方法;因为用这些语言编写代码比直接在 SMTLib 中编写代码更容易。
Python
from z3 import *
# Color enumeration
Color, (R, Y, B) = EnumSort('Color', ('R', 'Y', 'B'))
# An uninterpreted function for color assignment
col = Function('col', IntSort(), Color)
# Number of flowers
N = Int('N')
# Helper function to specify distinct valid choices:
def validPick (i1, i2, i3):
return And( Distinct(i1, i2, i3)
, 1 <= i1, 1 <= i2, 1 <= i3
, i1 <= N, i2 <= N, i3 <= N
)
# Helper function to count a given color
def count(pickedColor, flowers):
return Sum([If(col(f) == pickedColor, 1, 0) for f in flowers])
# Get a solver and variables for our assertions:
s = Solver()
f1, f2, f3 = Ints('f1 f2 f3')
# There's at least one flower of each color
s.add(Exists([f1, f2, f3], And(validPick(f1, f2, f3), col(f1) == R, col(f2) == Y, col(f3) == B)))
# You pick three, and at least one of them is red
s.add(ForAll([f1, f2, f3], Implies(validPick(f1, f2, f3), count(R, [f1, f2, f3]) >= 1)))
# You pick three, and at least one of them is yellow
s.add(ForAll([f1, f2, f3], Implies(validPick(f1, f2, f3), count(Y, [f1, f2, f3]) >= 1)))
# For every three flowers you pick, one of them has to be blue
s.add(ForAll([f1, f2, f3], Implies(validPick(f1, f2, f3), count(B, [f1, f2, f3]) == 1)))
# Find a satisfying value of N
print s.check()
print s.model()[N]
# See if there's any other value by outlawing the choice
nVal = s.model()[N]
s.add(N != nVal)
print s.check()
运行时,将打印:
sat
3
unsat
读取这个输出的方法是给定的条件在 ; 时确实是可满足的N=3
。正如你所寻求的那样。此外,如果您还断言N
is not 3
,那么就没有令人满意的模型,即,选择 of3
是由给定的条件所强制的。我相信这就是你想要建立的。
我希望代码清晰,但请随时要求澄清。如果您在 SMT-Lib 中确实需要它,您可以随时插入:
print s.sexpr()
在调用之前s.check()
,您可以看到生成的 SMTLib。
哈斯克尔
也可以在 Haskell/SBV 中编写相同的代码。请参阅此要点以获得几乎相同的文字编码:https ://gist.github.com/LeventErkok/66594d8e94dc0ab2ebffffe4fdabccc9请注意,Haskell 解决方案可以利用 SBV 的allSat
构造(返回所有令人满意的假设),更简单地表明这N=3
是唯一的解决方案。