在z3py中使用BitVec数组时,如何按数组顺序检索得到的解?

2024-09-28 01:31:50 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用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]中


Tags: inadd类型formodel顺序checkrange
1条回答
网友
1楼 · 发布于 2024-09-28 01:31:50

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

以与数组相同的顺序打印结果的最简单方法是: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]

相关问题 更多 >

    热门问题