
在某些计算场景中,我们需要遍历所有满足特定条件的二进制数值。例如,生成所有n位长、且恰好有m个置位(即1)的二进制数。原始问题中提供了一个python生成器函数bit_permutations,它能够高效地生成这些组合:
def trailing_zeros(v):
# 计算一个整数末尾0的个数
return (v & -v).bit_length() - 1
def bit_permutations(popcount, bits):
"""
生成所有bits位长、popcount个置位的二进制数值。
使用Gosper's Hack算法。
"""
if popcount < 0 or popcount > bits:
pass # 无效参数,不生成任何值
elif popcount == 0:
yield 0
elif popcount == bits:
yield (1 << bits) - 1
else:
# 初始值:最低popcount位全为1
v = (1 << popcount) - 1
while v < (1 << bits):
yield v
# Gosper's Hack: 找到下一个具有相同置位数量的组合
t = v | (v - 1)
v = (t + 1) | (((~t & -~t) - 1) >> (trailing_zeros(v) + 1))例如,bit_permutations(3, 5)会生成0b00111, 0b01011, 0b01101, ...等值。
然而,实际需求往往不仅限于获取这些原始数值,还需要它们的位反转形式。传统做法是为每个生成的数值调用一个单独的位反转函数,例如:
def reverse(v, bits):
"""
反转v的bits位。此实现限定bits <= 16,且效率不高。
"""
assert bits <= 16
v = ((v >> 1) & 0x5555) | ((v & 0x5555) << 1);
v = ((v >> 2) & 0x3333) | ((v & 0x3333) << 2);
v = ((v >> 4) & 0x0F0F) | ((v & 0x0F0F) << 4);
v = ((v >> 8) & 0x00FF) | ((v & 0x00FF) << 8);
return v >> (16 - bits)这种“先生成,再反转”的模式存在效率问题。每次迭代都需要额外的函数调用和位操作,尤其当bits值较大或需要生成大量组合时,性能开销会显著增加。此外,上述reverse函数对bits的限制(<=16)也限制了其通用性。
为了提高效率,我们可以将位反转的逻辑直接整合到bit_permutations生成器函数中。核心思想是:在生成每个原始数值v的同时,计算出其对应的位反转值reverse_v,并以元组(v, reverse_v)的形式一同返回。
def trailing_zeros(v):
# 计算一个整数末尾0的个数
return (v & -v).bit_length() - 1
def bit_permutations_with_reverse(popcount, bits):
"""
生成所有bits位长、popcount个置位的二进制数值,
并同时生成其位反转形式。
"""
if popcount < 0 or popcount > bits:
pass # 无效参数,不生成任何值
elif popcount == 0:
yield 0, 0 # 0的反转仍是0
elif popcount == bits:
# 所有位都是1,反转后仍是所有位都是1
all_ones = (1 << bits) - 1
yield all_ones, all_ones
else:
# 初始值:最低popcount位全为1
v = (1 << popcount) - 1
while v < (1 << bits):
# 关键改进:在生成v的同时计算reverse_v
# 使用Python的字符串格式化和切片反转,通用性强
reverse_v = int(format(v, f'0{bits}b')[::-1], 2)
yield v, reverse_v # 同时返回原始值和反转值
# Gosper's Hack: 找到下一个具有相同置位数量的组合
t = v | (v - 1)
v = (t + 1) | (((~t & -~t) - 1) >> (trailing_zeros(v) + 1))在优化后的函数中,计算reverse_v的核心代码是: reverse_v = int(format(v, f'0{bits}b')[::-1], 2)
这行代码分三步完成位反转:
这种方法简洁、通用性强,不受位宽限制(只要Python整数类型能表示即可),并且避免了复杂的位操作逻辑。
以下是如何使用优化后的bit_permutations_with_reverse函数的示例:
# 示例:生成5位长,3个置位的二进制数及其反转形式
popcount = 3
bits = 5
print(f"生成 {bits} 位长,{popcount} 个置位的二进制组合及其反转值:")
for perm, reverse_perm in bit_permutations_with_reverse(popcount, bits):
print(f"原始值: {format(perm, f'0{bits}b')} (十进制: {perm}), "
f"反转值: {format(reverse_perm, f'0{bits}b')} (十进制: {reverse_perm})")
# 预期输出示例:
# 原始值: 00111 (十进制: 7), 反转值: 11100 (十进制: 28)
# 原始值: 01011 (十进制: 11), 反转值: 11010 (十进制: 26)
# ...本文介绍了一种优化策略,用于高效生成指定位宽和置位数量的二进制组合,并同时获取其位反转形式。通过将位反转逻辑直接整合到生成器函数内部,我们实现了单次遍历即可获取原始值和反转值的目标,显著提升了代码的简洁性和执行效率。这种方法利用Python内置的字符串格式化和切片功能,提供了良好的通用性和可读性,适用于大多数场景。在对性能有极致要求的特定场景下,可以进一步探索更底层的位操作优化方案。
以上就是高效生成指定位宽和置位数量的二进制组合及其反转值的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号