bitset(C++实现)

简介: bitset(C++实现)

1. bitset类

1.1 私有成员

位图实际上就是一个指定比特位个数的连续内存空间,所以可以用STL内置的容器vector管理,除此之外,理论上任何类型都可以作为元素的类型,只不过为了容易理解,它的每个元素的类型被设定为char。

template<size_t N>    // N个比特位  
class bitset
{
private:
    vector<char> _bits; // 以char为单位管理
};

1. bitset接口

1.1 构造函数

构造一个有N位的位图,并将所有位初始化为0。

由于申请空间时的最小单位是char(1字节),也就是8个比特位。那么对于N个比特位,需要申请N/8+1个char型空间,因为N可能不是8的整数倍。

申请N=20个比特位,需要20/8+1=3个char。

bitset()
{
    _bits.resize(N / 8 + 1, 0);
}

1.2 set

要把某个比特位变成1,首先要知道这个比特位在哪个位置。

  1. 首先计算这个比特位在整个位图中的第x个char;
  2. 然后计算比特位在这个char的第y个位置;
  3. 最后将1左移y位和第x个char运算。

void set(size_t pos)
{
    assert(pos < N);
    size_t x = pos / 8;   // 在x个char
    size_t y = pos % 8;   // 在这个char的第y个比特位
    _bits[x] |= (1 << y); // 将这个比特位设为1
}

1.3 reset

要把某个比特位清空,即恢复到0状态,首先要找到它的位置,和set的操作是一样的,只有最后一步不同:

  1. 首先计算这个比特位在整个位图中的第x个char;
  2. 然后计算比特位在这个char的第y个位置;
  3. 最后将1左移y位后,整体取反,再和第x个char运算。

注意:

使用~对左移后的1按位取反,而不是逻辑取反!

void reset(size_t pos)
{
    assert(pos < N);
    size_t x = pos / 8;     // 在x个char
    size_t y = pos % 8;     // 在这个char的第y个比特位
    _bits[x] &= (~(1 << y));  // 将这个比特位设为0
}

1.4 flip

要把某个比特位取反,首先要找到它的位置:

  1. 首先计算这个比特位在整个位图中的第x个char;
  2. 然后计算比特位在这个char的第y个位置;
  3. 最后将1左移y位后,再和第x个char异或运算。
void flip(size_t pos)
{
    assert(pos < N);
    size_t x = pos / 8;     // 在x个char
    size_t y = pos % 8;     // 在这个char的第y个比特位
    _bits[x] ^= (1 << y);   // 将这个比特位取反
}

1.5 test

同样地:

  1. 首先计算这个比特位在整个位图中的第x个char;
  2. 然后计算比特位在这个char的第y个位置;
  3. 获取某个位的状态:
  • true:已被设置;
  • false:未被设置。
void test(size_t pos)
{
    assert(pos < N);
    size_t x = pos / 8;     // 在x个char
    size_t y = pos % 8;     // 在这个char的第y个比特位
    if (_bits[x] & (1 << y))  // 该位已被设置
        return true;
    else
        return false;
}

1.6 size、count

  • size:获取位图中可以容纳的位的个数。

直接返回模板参数N。

size_t size()
{
    return N;
}
  • count:获取位图中被设置的位的个数。

要知道位图中被设置为1的位的个数,就是遍历整个位图,统计1的个数。

  1. 将当前数x与x-1与运算得到新的数x;
  2. x是否为0:
  1. 为0:结束;
  2. 不为0:重复操作。

原因是每进行一次操作1,都会将数x最右端的1消去。那么在x不为0之前,进行了几次消1操作就是位图中1个个数。

size_t count()
{
    size_t count = 0;
    for (auto e : _bits)
    {
        char num = e;
        while (num)
        {
            num &= num - 1;
            count++;
        }
    }
    return count;
}

1.7 any、none、all

  • any:判断位图中是否有任何位被设置。

只需要遍历位图中所有的比特位,一旦有1则返回true,否则返回0。但也可以不用这么干,因为一个char里只要有一个不为0,整个char就是其他字符。ASCII=0对应的字符时'\0',但是它一般用在字符串处理中,个人认为在这里还是将0强转为char比较合适。

bool any()
{
    for (auto e : _bits)
    {
        if (e != (char)0)
            return true;
    }
    return false;
}
  • none:判断位图中是否全部位都没有被设置。

直接复用any的代码,它们是互斥的。

bool none()
{
    return !any();
}
  • 判断位图中是否全部位都被设置。

判断全部位置都被设置为1,分为两步:

  1. 判断所有完整的char中8个比特位是否都为1;
  2. 判断剩下(可能)不完整的char的所有比特位是否都为1。

同样地,第一步中可以判断这个char是否等于比特位全为1对应的字符,也就是(char)127,遍历剩下的比特位是否为1即可。

bool all()
{
    size_t size = _bits.size();
    for (size_t i = 0; i < size - 1; i++) // 前size-1个完整的char
    {
        if (_bits[i] != (char)127)
            return false;
    }
    for (size_t i = 0; i < N % 8; i++)    // 最后一个char的剩下位
    {
        if ((_bits[size - 1] & (1 << i)) == (char)0)
            return false;
    }
    return true;
}

1.8 打印函数

这不是bitset内置的成员函数,因为STL中重载了流输入>>和流输出<<运算符,可以直接打印bitset中的内容,而模拟实现并未重载它们,所以用打印函数输出容器内容。

思路和上一个很类似,也是分批打印:

void Print()
{
    string ret = "";
    size_t size = _bits.size();
    for (size_t i = 0; i < size - 1; i++)
    {
        for (size_t j = 0; j < 8; j++)
        {
            if (_bits[i] & (1 << j))
                ret += "1";
            else
                ret += "0";
        }
    }
    for (size_t j = 0; j < N % 8; j++)
    {
        if (_bits[size - 1] & (1 << j))
            ret += "1";
        else
            ret += "0";
    }
    cout << ret << endl;
}

2. bitset测试

void bitset_test1()
{
  xy::bitset<30> bs1;
  bs1.set(8);
  bs1.set(9);
  bs1.set(7);
  bs1.set(27);
  bs1.set(20);
  bs1.Print();
  cout << bs1.count() << endl;
  bs1.reset(8);
  bs1.reset(9);
  bs1.reset(20);
  bs1.Print();
  cout << bs1.count() << endl;
  cout << bs1.any() << endl;
  bs1.reset(7);
  bs1.reset(27);
  bs1.Print();
  cout << bs1.none() << endl;
}

输出:

000000011100000000001000000100
5
000000010000000000000000000100
2
1
000000000000000000000000000000
1

源代码

目录
相关文章
|
5月前
|
存储 算法 C++
c++ vector数组详细介绍(二)
c++ vector数组详细介绍(二)
190 0
|
4月前
|
存储 安全 算法
C++的内置数组和STL array、STL vector
C++的内置数组和STL array、STL vector
|
12月前
|
算法 固态存储 数据处理
位图(bitset)的使用【STL】
位图(bitset)的使用【STL】
47 1
|
5月前
|
存储 算法 数据安全/隐私保护
C++ 位运算 std::bitset类的使用介绍
C++ 位运算 std::bitset类的使用介绍
114 0
|
5月前
|
存储 算法 C++
c++ vector数组详细介绍(三)
c++ vector数组详细介绍(三)
80 0
|
5月前
|
存储 安全 算法
c++ vector数组详细介绍(一)
c++ vector数组详细介绍(一)
210 0
|
5月前
std::vector不隐式拷贝进行添加元素
std::vector不隐式拷贝进行添加元素
|
存储 API C++
C++ std::vector元素的内存分配问题
在使用C++ STL的vector时,下面三种写法有什么不同呢?其内存分配是怎么样的呢? 首先,说结论吧(假设T是一个定义好的类): 对于std::vector<T> vec;这种方式vec在栈上(stack),而其中的元素T保存在堆上(heap); 对于std::vector<T>* vec = new std::vector<T>();这种方式vec和其中的元素T都保存在堆上; 对于std::vector<T*> vec;这种方式vec在栈上(stack),而其中的元素T保存在堆上(heap);和第一种情况类似。
155 0
|
存储
BitSet—位图
BitSet—位图
|
存储 C++ 容器
【C++】-- STL之vector详解(一)
【C++】-- STL之vector详解
134 0
【C++】-- STL之vector详解(一)