首先来看一下linux内核代码,如下:
#define BIT_PER_LONG (CHAR_BIT * sizeof(unsigned long))
#define BIT_MASK(idx) (1UL << ((idx) % BIT_PER_LONG))
#define BIT_WORD(idx) ((idx) / BIT_PER_LONG)
/* Helpers */
static inline void __set_bit(unsigned idx, unsigned long *bmap)
{
bmap[BIT_WORD(idx)] |= BIT_MASK(idx);
}
乍一看一脸懵逼,思考之后发现这个接口的作用就是将idx位置1。为什么呢?
下面看一个简单的例子:
#define MASK 8
void add_to_bitarray(char *bitarr, int num){ /* num代表要插进数组中的数 */
bitarr[num / MASK] |= (1 << (num % MASK));
}
char字符数组即char *bitarr中一个char可以表示8位置。若数组只有两个字符,那么bitarr[1],在内存中只有2乘以8 = 16位。
1)首先,num/MASK表示的是数组中的第几位,比如数组要代表17,那么num = 16,17/MASK = 2。即代表数组的第2位。
2)接着,num % MASK代表一个字节中的第几位,17 % 8 = 1,即表示通过第二位来表示。
3)bitarr[num / MASK] |= (1 << (num % MASK)),表示的是将bitarr[1]中的第二位置1。
4)bitarr[1]中的第二位置1,即表示的是第17位为1。
接着如何判断一个整数是否在二进制数中呢?
int is_in_bitarray(char *bitarr, int num){
return bitarr[num / MASK] & (1 << (num % MASK));
}
如何删除二进制数组中的一个整数?
void clear_bitarray(char *bitarr, int num){
bitarr[num >> SHIFT] &= ~(1 << (num & MASK));
}