编程珠玑-位向量

如何使用位逻辑运算（如与、或、移位）来实现位向量？

一、疑问

• 什么是位向量？

位向量（bit vector）就是由一些二进制位组成的向量。

例如，可以用如下字符串来表示集合{1, 2, 3, 5, 8, 13}：
0 1 1 1 0 1 0 0 1 0 0 0 0 1 0 0 0 0 0 0
代表集合中数值的位置为1，其他所有的位置都置为0。

使用一个大数组，令N=1000000，若数组类型为int,则数组定义为int a[N/32+1]。
显然a[0]可以表示0至31的整数，以此类推，a[i]可以表示32i至32(i+1)-1的整数。

• 移位运算的几个规则

m/2^n=m>>n
m%(2^n)=m&(2^n-1)
将变量a的第k位置1: a = ( a | (1 << k) )
将变量a的第k位清0: a = ( a & ~(1 << k) )

二、实现

#define BITSPERWORD 32      // 一个int数字32位
#define shift 5             // 2^5 = 32 用于判断是第几个32位int数
#define mask 0x1F           // 0b11111 , (2^5-1 = 31)
#define n 10000000          // 位向量元素个数
int a[1 + n / BITSPERWORD]; // 1 + n / BITSPERWORD = int数的个数；+1实现的进位制度

// 加入位向量
void set(int i)
{
    a[i >> shift] |= (1 << (i & mask));
    /*    
    i>>shift 等价于m/2^n=m>>n，即 i/32 。
    a类似一个二维数组，行是每一个int里的每一个位列是a数组里的每一个int。即判断 i 映射到位向量里的第几个int里。

    i & mask 等价于m%(2^n)=m&(2^n-1)，即 i%32 。
    
    将变量a的第k位置1: a = ( a | (1 << k) )
    1<< (i & mask) 将 1 向左移动 i 后五位所表示的大小，也就是i在位上的位置，再或运算，将 i 映射在a[i /32 ]上的位置为1
    */
}

void clr(int i)
{
	// 与set类似，将变量a的第k位清0: a = ( a & ~(1 << k) )
    a[i >> shift] &= ~(1 << (i & mask));
}

int test(int i)
{
    return a[i >> shift] & (1 << (i & mask));
}