在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]如果我希望查找到的结果中,把结果中包含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]这一种情况
如果我希望符合条件的结果中,把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语句来添加条件,实现全部符合条件的模型结果的获取。