【python环境下Z3约束求解器学习笔记】And和Or的用法

在Z3约束求解器中，我们可能需要寻找同时满足两个条件的模型，也可能需要寻找满足两个条件中一个条件的模型，这个时候，我们可以借助And方法和Or方法来实现

引入

我们先来举一个简单的例子

当我们想要查找满足a+b=2且a、b均为正整数时，a、b可能的值
我们可以通过如下方法查找到满足条件的结果

from z3 import *

a = Int('a')
b = Int('b')
s = Solver()
s.add(a+b<2)
s.add(a>=0)
s.add(b>=0)

s.check()
print(s.model())

这个时候，符合条件的结果总共有四种（model()方法只会输出一种，所以在控制台上我们只能看到一组a和b的值）

[b = 0, a = 0]
[b = 0, a = 1]
[b = 1, a = 0]
[b = 1, a = 1]

And语句

如果我希望查找到的结果中，把结果中包含a=1的情况和包含b=1情况去掉，只保留[b = 0, a = 0]这一组，这个时候我们可以通过两个add语句实现

s.add(a!=1)
s.add(b!=1)

也可以借助And语句实现

s.add(And(a!=1,b!=1))

下面是借助And语句实现的完整代码

from z3 import *

a = Int('a')
b = Int('b')
s = Solver()
s.add(a+b<2)
s.add(a>=0)
s.add(b>=0)
s.add(And(a!=1,b!=1))
s.check()
print(s.model())

此时，符合条件的结果只有[b = 0, a = 0]这一种情况

Or语句

如果我希望符合条件的结果中，把a为1且b为1的情况去掉，也就是去掉[b = 1, a = 1]这一组，这个时候我们不能使用s.add(a!=1)和s.add(b!=1)来实现这一操作，可以借助Or语句来实现

s.add(Or(a!=1,b!=1))

下面是借助Or语句实现的完整代码

from z3 import *

a = Int('a')
b = Int('b')
s = Solver()
s.add(a+b<2)
s.add(a>=0)
s.add(b>=0)
s.add(Or(a!=1,b!=1))
s.check()
print(s.model())

这种情况下，符合条件的结果中，不存在a和b同时为1的情况了，只有a、b不同时为1的三种结果。

技巧

关于Or还有一个很实用的例子

a = Int('a')
b = Int('b')
s = Solver()
s.add(1 <= a)
s.add(a <= 20)
s.add(1 <= b)
s.add(b <= 20)
s.add(a >= 2*b)
while s.check() == sat:
  print (s.model())
  s.add(Or(a != s.model()[a], b != s.model()[b]))

因为model()方法只能输出最后一个check()方法的模型，导致我们无法直接获取到全部符合条件的模型
因此，借助while循环和Or语句来添加条件，实现全部符合条件的模型结果的获取。