使用位逻辑运算实现位向量所指实现位向量的设置,清零以及探测功能。
#define BITPERWORD 32 #define SHIFT 5 #define MASK 0x1F #define N 10000000 int a[1+N/BITPERWORD] //设置数组第i位为1 void set(int i) { a[i>>SHIFT] |= (1<<(i&MASK)); } //清空数组第i位为0 void clr(int i) { a[i>>SHIFT] &= ~ (1<<(i&MASK)); } //查询数组第i位数字 void test(int i) { return a[i>>SHIFT] &(1<<(i&MASK)); }
void set(int i) { a[i/32] |= (1<<(i%32)); }
void clr(int i) { a[i/32] &= ~ (1<<(i%32)); }3.test算法中通过1<<(i&MASK),其中i&MASK是计算需要左移的位数,然后把1向左移动这么多位,然后把和数组中到元素与运算,数组中元素为1就返回1,数组中元素为0就返回0.test算法可以改写为:
void test(int i) { return a[i/32] &(1<<(i%32)); }