如何实现简单的位数组(bit array)

在 comp.lang.c 上面看到一则不错的 FAQ，《How can I implement sets or arrays of bits？》感觉很实用，仅仅使用了几个简单的宏就实现了一个基本的位数组（bitset）。

#include "limits.h"#defineBITMASK(b) (1 < ((b) % CHAR_BIT))#defineBITSLOT(b) ((b) / CHAR_BIT)#defineBITSET(a, b) ((a)[BITSLOT(b)] |= BITMASK(b))#defineBITCLEAR(a, b) ((a)[BITSLOT(b)] &= ~BITMASK(b))#defineBITTEST(a, b) ((a)[BITSLOT(b)] & BITMASK(b))#defineBITNSLOTS(nb) ((nb + CHAR_BIT - 1) / CHAR_BIT)

下面是一些简单的例子：

• 声明一个固定长度（50个bit）的位数组：

charbitarray[BITNSLOTS(50)];

• 设置位数组中的某一位：

BITSET(bitarray,23);

• 检测某一位

if(BITTEST(bitarray,35)) ...

• 求两个位数组的并集

for(i =0; i < BITNSLOTS(47); i++)array3[i]= array1[i] | array2[i];

• 求两个位数组的交集

for(i =0; i < BITNSLOTS(47); i++)array3[i]= array1[i] & array2[i];

下面是一个完整的例子，利用Sieve of Eratosthenes算法求素数：

#include#include#defineMAX 10000intmain(){charbitarray[BITNSLOTS(MAX)];inti, j;memset(bitarray,0, BITNSLOTS(MAX));for(i =2; i < MAX; i++){if(!BITTEST(bitarray, i)){printf("%d", i);for(j = i + i; j < MAX; j +=i)BITSET(bitarray, j);}}return0;}