文章目录
- 0. 位运算 基本介绍
- 1. 位运算基本使用 + 连带题目
- 191.位1的个数
- 338.比特位计数
- 461.汉明距离
- 136.只出现一次的数字
- 260.只出现一次的数字III
- 2. 使用位运算解决算法题
- 面试题01.01.判定字符是否唯一
- 371.两整数之和
- 137.只出现一次的数字II
- 面试题17.04.消失的数字
- 面试题17.19.消失的两个数字
0. 位运算 基本介绍
我们知道,一般基本位运算分为以下几种:
- & 按位与:有0则为0
- | 按位或:有1则为1
- ^ 按位异或:记法有2
- 相同为0,相异为1
- 不进位相加(00->0, 01->1, 11->0,相当于两数相加但不进位)
- ~ 按位取反:对操作数按位取反,即0变为1,1变为0。
- << 向左移动:相当于将其乘以2的若干次方
- >> 向右移动:相当于将其除以2的若干次方
1. 位运算基本使用 + 连带题目
当我们理解了基础位运算,我们要确保可以解决下图红字中的基本问题:
通过上图设计的题目和思路,我们可以顺势解决以下几道题:
191.位1的个数
思路:
如图,我们只需要 将n的每一位都&1 即可:
- 将n&1,结果更新到ret
- n右移一位,持续此过程直到n为0
代码:
int hammingWeight(uint32_t n) {int ret = 0; // 记录结果while (n){ret += n & 1;n >>= 1;}return ret;
}
338.比特位计数
思路:
题目要求计算一个(1~n)范围内每个元素,二进制中1的个数 :
- 创建结果数组ret,变量count记录当前元素二进制中1的个数
- 将整个过程放到一个从0~n的循环中
- 计算当前元素位一的个数
- 将count给到ret,直至循环结束,返回ret
代码:
vector<int> countBits(int n) {vector<int> ret(n+1, 0);for(int i = 0; i <= n; ++i) // 统计0 ~ n中每个元素二进制位一的个数{int count = 0;int tmp = i;while(tmp) // 计算tmp中1的个数{count += tmp & 1;tmp >>= 1;}// 存储其1的个数ret[i] = count;}return ret;}
461.汉明距离
思路:
如图所示,我们知道异或相当于(相同为0,相异为1),只需要统计两数异或之后的结果中1的个数即就是不同位的个数,即汉明距离。
代码:
int hammingDistance(int x, int y) {int ret = 0;int tmp = x ^ y;// 异或后 1的个数 即为 二进制位不同位置数while(tmp){ret += tmp & 1;tmp >>= 1;}return ret;
}
136.只出现一次的数字
思路:
题目要求找到整数数组中只出现过一次的数字,此题可以用哈希表解题,但位运算的时间空间复杂度总体更为优秀。
- 我们知道,a ^ a = 0; b ^ 0 = b;
- 根据该思路,将数组所有元素异或,剩余元素即为只出现一次的。
代码:
int singleNumber(vector<int>& nums) {int ret = 0;// 将全部元素异或,结果为只出现一次的数字for(int i : nums){ret ^= i;}return ret;
}
260.只出现一次的数字III
思路:
- 首先异或所有元素,得到tmp(所求两数的异或结果)
- 将tmp & -tmp 得到两元素的不同位differ
- 根据此不同位进行划分:
- 如果当前元素num与differ按位与的结果不等于0,则a ^= num
- 反之, b ^= num
- 最后a,b即为两个只出现一次的数字。
代码:
vector<int> singleNumber(vector<int>& nums) {int tmp = 0;for(int num : nums) tmp ^= num; // 全部元素异或// 防止溢出int differ = (tmp == INT_MIN ? tmp : tmp & (-tmp)); // 找a,b的不同位int a = 0, b = 0;for(int num : nums){ // 根据条件划分if((num & differ) != 0)a ^= num;elseb ^= num;}return {a, b};
}
2. 使用位运算解决算法题
上面的题是小试牛刀,下面的题正式进行位运算算法的代码编写。
面试题01.01.判定字符是否唯一
思路:
上题解法利用位图思想 :我们通过将字符串的每个元素存到int型变量bitMap
中(该位为0:未出现过,为1:出现过),通过判断所有位是否有1则可判断字符串的字符是否唯一。
代码:
bool isUnique(string astr) {// 位图if(astr.size() > 26) return false;int bitMap = 0; // 从后向前存放字母的出现次数,0代表未出现过,1代表出现过for(char ch : astr){int i = ch - 'a'; // 取该字符位if((bitMap >> i) & 1 == 1) return false; // 如果这一位存在,falsebitMap |= (1 << i);}return true;
}
371.两整数之和
题意很清晰:即不使用±运算符计算两个整数的和。
思路:
我们知道:
- 按位异或^ = 不进位相加,而我们需要相加的结果,则只需要找到进位即可。
- 而(a & b)就是进位结果,但我们进位是向前一位进位,所以进位为(a & b) << 1。重复上述步骤,直到进位为0,就得到了最终结果。
代码:
int getSum(int a, int b) {while(b != 0){int nsum = a ^ b; // 不进位相加结果int carry = (a & b) << 1; // 进位数a = nsum;b = carry; // 重复步骤直到b==0}return a;
}
137.只出现一次的数字II
思路:
-
定义一个
vector<int> count(32)
,用于存储nums中所有元素的二进制表示的各个位的出现次数; -
对于记录完毕的count,count的每一位都有如下四种情况:
- 如图所示,即0, 1, 3n, 3n + 1四种情况
- 而对这四种情况模3后结果有两种情况:0 和 1
-
将得出的结果还原到ret中(用|=还原)
代码:
int singleNumber(vector<int>& nums) {vector<int> count(32); // 存放数组所有元素的位for(int num : nums)for(int i = 0; i < 32; ++i) // 存{count[i] += (num >> i) & 1;}
// 遍历count中的32个元素,还原只出现一次的元素到ret中int ret = 0;for(int i = 0; i < 32; ++i){ret |= (count[i] % 3) << i;}return ret;
}
面试题17.04.消失的数字
思路:
我们知道:a ^ a = 0 且 0 ^ b = b ,则利用这个性质,将nums中的数与0~n的所有数异或,最终结果则为消失的数字。
代码:
int missingNumber(vector<int>& nums) {// 位运算int n = nums.size();int miss = 0;for(int i = 0; i < n; ++i) // 与数组所有元素and 0~n-1的元素异或{miss ^= i ^ nums[i];}miss ^= n; // 异或nreturn miss;
}
面试题17.19.消失的两个数字
思路:
“只出现一次的数字Ⅲ” 中有个思想就是通过两元素的不同位,进行划分,这里也是一样
- 异或所有数:得到缺失的两个数字的异或和
- 获取 tmp 中最右边为 1 的位数:找到两个缺失数字的不同位
- 根据 differ 的值,将数组元素划分为两组进行异或操作
代码:
vector<int> missingTwo(vector<int>& nums) {int tmp = 0;int n = nums.size() + 2; // 数组中缺失两个数字,所以数组长度为n+2// 1. 异或所有数for (int i = 1; i <= n; ++i)tmp ^= i;for (int num : nums)tmp ^= num;// 2. 获取最右边为1的位数(不同位)int differ = tmp & -tmp;// 3. 根据differ,划分数组元素进行异或int a = 0, b = 0;for (int i = 1; i <= n; ++i) {if (i & differ)a ^= i;elseb ^= i;}for (int num : nums) {if (num & differ)a ^= num;elseb ^= num;}return {a, b};
}