Flip Bits
Question
- lintcode: Flip Bits
Problem Statement
Determine the number of bits required to flip if you want to convert integer n to integer m.
Notice
Both n and m are 32-bit integers.
Example
Given n = 31
(11111), m = 14
(01110), return 2
.
题解
比较两个数不同的比特位个数,显然容易想到可以使用异或处理两个整数,相同的位上为0,不同的位上为1,故接下来只需将异或后1的个数求出即可。容易想到的方法是移位后和1按位与得到最低位的结果,使用计数器记录这一结果,直至最后操作数为0时返回最终值。这种方法需要遍历元素的每一位,有咩有更为高效的做法呢?还记得之前做过的 O1 Check Power of 2 吗?x & (x - 1)
既然可以检查2的整数次幂,那么如何才能进一步得到所有1的个数呢?——将异或得到的数分拆为若干个2的整数次幂,计算得到有多少个2的整数次幂即可。
以上的分析过程对于正数来说是毫无问题的,但问题就在于如果出现了负数如何破?不确定的时候就来个实例测测看,以-2为例,(-2) & (-2 - 1)的计算如下所示(简单起见这里以8位为准):
11111110 <==> -2 -2 <==> 11111110
+ &
11111111 <==> -1 -3 <==> 11111101
= =
11111101 11111100
细心的你也许发现了对于负数来说,其表现也是我们需要的——x & (x - 1)
的含义即为将二进制比特位的值为1的最低位置零。逐步迭代直至最终值为0时返回。
C/C++ 和 Java 中左溢出时会直接将高位丢弃,正好方便了我们的计算,但是在 Python 中就没这么幸运了,因为溢出时会自动转换类型,Orz... 所以使用 Python 时需要对负数专门处理,转换为求其补数中0的个数。
Python
class Solution:
"""
@param a, b: Two integer
return: An integer
"""
def bitSwapRequired(self, a, b):
count = 0
a_xor_b = a ^ b
neg_flag = False
if a_xor_b < 0:
a_xor_b = abs(a_xor_b) - 1
neg_flag = True
while a_xor_b > 0:
count += 1
a_xor_b &= (a_xor_b - 1)
# bit_wise = 32
if neg_flag:
count = 32 - count
return count
C++
class Solution {
public:
/**
*@param a, b: Two integer
*return: An integer
*/
int bitSwapRequired(int a, int b) {
int count = 0;
int a_xor_b = a ^ b;
while (a_xor_b != 0) {
++count;
a_xor_b &= (a_xor_b - 1);
}
return count;
}
};
Java
class Solution {
/**
*@param a, b: Two integer
*return: An integer
*/
public static int bitSwapRequired(int a, int b) {
int count = 0;
int a_xor_b = a ^ b;
while (a_xor_b != 0) {
++count;
a_xor_b &= (a_xor_b - 1);
}
return count;
}
};
源码分析
Python 中 int 溢出时会自动变为 long 类型,故处理负数时需要求补数中0的个数,间接求得原异或得到的数中1的个数。
考虑到负数的可能,C/C++, Java 中循环终止条件为a_xor_b != 0
,而不是a_xor_b > 0
.
复杂度分析
取决于异或后数中1的个数,O(max(ones in a ^ b))
.
关于 Python 中位运算的一些坑总结在参考链接中。