在z3py中使用BitVec数组时，如何按数组顺序检索获得的解决方案?

Question

我正在尝试使用 z3 中的数组类型解决问题。 因为我需要使用BitVec类型，所以我将数组声明为：

numbers = [BitVec(chr(i), 8) for i in range(0, 4)]

然后：

s = Solver()
s.add(numbers[0] == 100)
s.add(numbers[1] == numbers[0] + 2)
s.add(numbers[3] == numbers[1] + numbers[0])
s.add(numbers[2] == numbers[1] - 4)
print(s.check())
print(s.model())

输出：

sat
[ = 98,  = 202,  = 102,  = 100]

但是它不会按顺序打印结果，有没有办法按顺序打印它们？

例：

[ = 100, = 102, = 98, = 202 ]

我还有另一个疑问。有没有办法对数字的频率设置限制：

numbers = [BitVec(chr(i), 8) for i in range(0, 4)]
s = Solver()
s.add(numbers[0] == 100)
s.add(numbers[0] + numbers[1] + numbers[2] == 200)
s.add(collections.Counter(numbers)[100] == 1) # something like that
print(s.check())
print(s.model())

要设置该数字，100 必须仅存在于数字 [0] 中。

Answer 1

对于您的第一个问题，我首先建议为变量提供一些可读的名称。chr(i) for i in range(0, 4)是 4 个不可打印的字符。最好对数字 0..9 使用chr(i+48)，对字母 A 使用chr(i+65)更好。Z

以与数组相同的顺序打印结果的最简单方法只是：print ([s.model[n].as_long() for n in numbers]).

对于你的第二个问题，我建议使用 Z3 的Sum函数。例如：

numbers = [BitVec('N'+chr(i+48), 8) for i in range(0, 4)]
s = Solver()
s.add(numbers[0] == 100)
s.add(numbers[0] + numbers[1] + numbers[2] == 200)
s.add(Sum([If(n == 100, 1, 0) for n in numbers]) == 1)
print(s.check())
m = s.model()
print (m)
print ([m[n].as_long() for n in numbers])

哪些输出(在我的测试用例中(：

sat
[N3 = 116, N1 = 106, N0 = 100, N2 = 250]
[100, 106, 250, 116]