Bloom filter是一种空间效率很高的概率性数据结构,用来判断一个元素是否属于一个集合。它通过使用多个哈希函数和一个位数组来实现。
以下是Bloom filter的使用步骤:
初始化Bloom filter:创建一个位数组,所有位都初始化为0。
添加元素:对于要添加的元素,使用多个哈希函数生成多个哈希值。然后将对应的位数组位置设为1。
查询元素:对于要查询的元素,使用同样的哈希函数生成多个哈希值。检查对应的位数组位置是否都为1。如果都为1,则可能在集合中;如果有任何一个位为0,则肯定不在集合中。
以下是一个简单的Python代码示例,演示如何使用Bloom filter:
import hashlib
from bitarray import bitarray
class BloomFilter:
def __init__(self, size, num_hash):
self.size = size
self.num_hash = num_hash
self.bit_array = bitarray(size)
self.bit_array.setall(0)
def add(self, element):
for i in range(self.num_hash):
index = self._hash(element, i)
self.bit_array[index] = 1
def contains(self, element):
for i in range(self.num_hash):
index = self._hash(element, i)
if self.bit_array[index] == 0:
return False
return True
def _hash(self, element, index):
hash = hashlib.sha256()
hash.update(bytes(element, 'utf-8'))
return int.from_bytes(hash.digest(), 'little') % self.size
# 使用示例
bloom_filter = BloomFilter(100, 3)
bloom_filter.add('apple')
bloom_filter.add('banana')
print(bloom_filter.contains('apple')) # True
print(bloom_filter.contains('banana')) # True
print(bloom_filter.contains('orange')) # False
在实际使用中,你可以根据需要调整位数组的大小和哈希函数的数量,以平衡空间和查询准确性的需求。