位元運算

Wed, 23 Mar 2022 07:48:37 +0800

最近迷上有趣的位元運算，發現原來有很多可玩的地方，所以就簡單紀錄一下這裡幾乎都是以 32 bit 為主，若 64 bit 請自行擴充

ABS

取絕對值

int abs(int n) {
 return ((n >> 31) ^ n) - (n >> 31);
}

Count 1

計算 1 存在的數量

int count_1(uint32_t n) {
 n = (n & 0x55555555) + ((n & 0xAAAAAAAA) >> 1);
 n = (n & 0x33333333) + ((n & 0xCCCCCCCC) >> 2);
 n = (n & 0x0F0F0F0F) + ((n & 0xF0F0F0F0) >> 4);
 n = (n & 0x00FF00FF) + ((n & 0xFF00FF00) >> 8);
 n = (n & 0x0000FFFF) + ((n & 0xFFFF0000) >> 16);
 return n;
}

或用 Brian Kernighan's Algorithm，此算法主要是由觀察 n 與 n-1 的 & 會把最小非零位元歸零 x..x10..0 & x..x01..1 = x..x00..0

int count_1(uint32_t n) {
 int distance = 0;
 while (n != 0) {
 distance += 1;
 // remove the rightmost bit of '1'
 n &= n - 1;
 }
 return distance;
}

Pow of 2

簡單來說就是檢查 Count 1 == 1，所以可以用前面的方法。

但 Brian Kernighan's Algorithm 的想法延伸可以找到更簡單作法， n 與 n-1，如果 n 呈現為 xx...x10...0，而 n-1 為 xx..x01...1，相 & 後為 xx...x00...0，及 n 的最低位非 0 以下都會被壓縮成 0。

bool isPowerOfTwo(uint32_t n){
 return (n & (n-1)) == 0;
}

Swap

可以完成 in-placed swap

void swap(int *a, int *b) {
 *a ^= *b;
 *b ^= *a;
 *a ^= *b;
}

Reverse Bits

將數字前後顛倒

uint32_t reverseBits(uint32_t n) {
 n = ((n & 0xFFFF0000) >> 16) | ((n & 0x0000FFFF) << 16);
 n = ((n & 0xFF00FF00) >> 8) | ((n & 0x00FF00FF) << 8);
 n = ((n & 0xF0F0F0F0) >> 4) | ((n & 0x0F0F0F0F) << 4);
 n = ((n & 0xCCCCCCCC) >> 2) | ((n & 0x33333333) << 2);
 n = ((n & 0xAAAAAAAA) >> 1) | ((n & 0x55555555) << 1);
 return n;
}

Find not duplicate / find difference

檢查字串裡沒有重複的字元 / 檢查兩字串的字元差(確定只有一個了話)。

透過 xor 自己會得 0 的性質。

Coding | Chia-An Lee