【python环境下Z3约束求解器入门学习教程】基础教程

1,237 阅读4分钟

一个简单的实例

问题描述:解不等式a+b<2,a>=0,b>=0的整数a和b

from z3 import *

a = Int('a')#定义一个整形 a
b = Int('b')#定义一个整形 b
s = Solver()#生成一个约束求解器
s.add(a+b<2)#添加约束条件 a+b<2
s.add(a>=0)#添加约束条件 a>=0
s.add(b>=0)#添加约束条件 b>=0

print(s.check())#检查约束求解器是否有解,如果有,返回sat; 如果不满足,返回unsat
print(s.model())#输出结果

执行结果: 在这里插入图片描述

  • 在Z3中,除了整形(Int),还可求取实数型(Real),浮点型(Float16、Float32、Float64、Float128),布尔型(Bool)等
  • 运算符有<,<=,>,> =,==和!=
  • 虽然这个不等式的结果有很多,但约束求解器只能求出一个结果,后续会讲到求解多种结果的方法

另一种求解方法:solver

问题描述:求解满足下列条件:a+b<2,a>=0,b>=0的整数a和b

from z3 import *

a = Int('a')#定义一个整形 a
b = Int('b')#定义一个整形 b

solve(a+b<2,a>=0,b>=0)#将a和b的约束条件写在括号里用逗号分割

执行结果: 在这里插入图片描述

这种方法虽然书写简单,但在约束条件太多的情况,不容易理解

Real类型的简单使用

问题描述:求解3x=1时,x的值

from z3 import *

x = Real('x')
solve(3*x == 1)

set_option(rational_to_decimal=True)
solve(3*x == 1)

执行结果: 在这里插入图片描述

Real类型的数据求解默认为分数形式,如果设置rational_to_decimal=True则结果位小数形式,默认显示小数点后10位,10位后用?表示后面还有

实数型小数保留问题:precision

问题描述:求解满足下列条件:a+b<2,a>=0,b>=0的整数a和b

from z3 import *

x = Real('x')
y = Real('y')
s = Solver( ) #生成一个约束求解器
s.add(x**2 + y**2 == 3) #添加约束条件:x的平方加y的平方等于3
s.add(x**3 == 2) #添加约束条件:x的三次方等于2

set_option(precision=30)#输出结果保留小数点后30位

print(s.check())#检查约束求解器是否有解,如果有,返回sat; 如果不满足,返回unsat
print(s.model())#输出结果

执行结果: 在这里插入图片描述

Not和Or的简单使用

公式简化:Simplify

问题描述:简化公式x + y + 2*x + 3

from z3 import *

x = Int('x')
y = Int('y')
print (simplify(x + y + 2*x + 3))
print (simplify(x < y + x + 2))
print (simplify(And(x + 1 >= 3, x**2 + x**2 + y**2 + 2 >= 5)))

执行结果: 在这里插入图片描述

主要用于对复杂公式的自动化简化

在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语句来添加条件,实现全部符合条件的模型结果的获取。