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,首先要知道这个比特位在哪个位置。
- 首先计算这个比特位在整个位图中的第x个char;
- 然后计算比特位在这个char的第y个位置;
- 最后将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的操作是一样的,只有最后一步不同:
- 首先计算这个比特位在整个位图中的第x个char;
- 然后计算比特位在这个char的第y个位置;
- 最后将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
要把某个比特位取反,首先要找到它的位置:
- 首先计算这个比特位在整个位图中的第x个char;
- 然后计算比特位在这个char的第y个位置;
- 最后将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
同样地:
- 首先计算这个比特位在整个位图中的第x个char;
- 然后计算比特位在这个char的第y个位置;
- 获取某个位的状态:
- 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的个数。
- 将当前数x与x-1与运算得到新的数x;
- x是否为0:
- 为0:结束;
- 不为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,分为两步:
- 判断所有完整的char中8个比特位是否都为1;
- 判断剩下(可能)不完整的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