python 位图数据结构适合检查整数是否存在

nongcunqq

于 2024-02-06 11:15:38 发布

阅读量247

点赞数 2

分类专栏：笔记 python 文章标签： python 数据结构开发语言

本文链接：https://blog.csdn.net/nongcunqq/article/details/136053770

版权

python 同时被 2 个专栏收录

335 篇文章 0 订阅

订阅专栏

笔记

320 篇文章 3 订阅

订阅专栏

原生位图举例

# 初始化一个位图
bitmap = 0

# 假设我们有一组数字
numbers = [2, 5, 8, 10]

# 将数字存在的位置设为1
for num in numbers:
    bitmap |= (1 << num)

# 检查数字是否存在
check_number = 5
if bitmap & (1 << check_number) != 0:
    print(f"{check_number} exists in the set.")
else:
    print(f"{check_number} does not exist in the set.")

使用 bitarray 库，如果不存在需安装
pip install bitarray

from bitarray import bitarray

# 初始化一个位图
bitmap = bitarray(200)  # 假设数字范围在0到9之间
bitmap.setall(0)

print(bitmap)
# 假设我们有一组数字
numbers = [2, 5, 8, 10, -1]

# 将数字存在的位置设为1
for num in numbers:
    bitmap[num] = 1

# 检查数字是否存在
check_number = -1
if bitmap[check_number]:
    print(f"{check_number} exists in the set.")
else:
    print(f"{check_number} does not exist in the set.")