c++如何实现一个简单的BitSet_c++位集数据结构与操作【源码】

C++手写BitSet通过uint32_t数组按位存储布尔值,支持set/reset/test/flip等操作;核心是将位位置映射到数组下标和位偏移,用位运算高效操作,不依赖标准库bitset。

用 C++ 实现一个简单的 BitSet,核心是用整型数组(如 uint32_tunsigned long)按位存储布尔状态,通过位运算高效操作。不需要依赖 ,自己手写更灵活、便于理解底层原理。

基本设计思路

把 N 个 bit 映射到长度为 (N + 31) / 32uint32_t 数组上(以 32 位为例):
- 第 i 个 bit 对应数组下标 i / 32,位偏移 i % 32
- 用 |= 置位,&= ~ 清位,& 查位
- 支持构造、置位(set)、清位(reset)、查询(test)、翻转(flip)等操作

完整可运行源码(C++11 及以上)

// BitSet.h

#include 
#include
#include

class BitSet {
private:
std::vector data;
size_t num_bits;

static constexpr size_t BITS_PER_WORD = 32;
size_t word_index(size_t pos) const { return pos / BITS_PER_WORD; }
size_t bit_offset(size_t pos) const { return pos % BITS_PER_WORD; }

public:
explicit BitSet(size_t n) : num_bits(n),
data((n + BITS_PER_WORD - 1) / BITS_PER_WORD, 0) {}

void set(size_t pos) {
if (pos >= num_bits) throw std::out_of_range("BitSet::set: index out of range");
data[word_index(pos)] |= (1U << bit_offset(pos));
}

void reset(size_t pos) {
if (pos >= num_bits) throw std::out_of_range("BitSet::reset: index out of range");
data[word_index(pos)] &= ~(1U << bit_offset(pos));
}

bool test(size_t pos) const {
if (pos >= num_bits) throw std::out_of_range("BitSet::test: index out of range");
return data[word_index(pos)] & (1U << bit_offset(pos));
}

void flip(size_t pos) {
if (pos >= num_bits) throw std::out_of_range("BitSet::flip: index out of range");
data[word_index(pos)] ^= (1U << bit_offset(pos));
}

size_t size() const { return num_bits; }
size_t num_words() const { return data.size(); }

// 可选:批量设置/清空全部
void set_all() {
for (auto& w : data) w = ~0U;
// 最后一个字可能越界,需掩码处理(略,可按需扩展)
}

void reset_all() {
data.assign(data.size(), 0);
}
};

使用示例

int main() {
BitSet bs(100); // 支持 0~99 共 100 个 bit

bs.set(5);
bs.set(10);
bs.flip(5); // 5 变成 0

std::cout << bs.test(5) << " "; // 输出 0
std::cout << bs.test(10) << "\n"; // 输出 1

return 0;
}

进阶建议(可选扩展)

- 支持模板参数指定字长(如 uint64_t),提升大容量性能
- 添加 count():用内置函数 __builtin_popcount(GCC/Clang)或查表法统计 1 的个数
- 实现 operator[] 返回代理类,支持 bs[5] = true 写法
- 增加迭代器,方便遍历所有置位位置(如找下一个 1 的位置)
- 加入边界检查开关(debug 模式开启,release 模式关闭)提升效率

基本上就这些。手写 BitSet 不复杂但容易忽略越界和字对齐细节,抓住“分块+位偏移”这个关键,就能稳稳落地。