题目
请实现一个函数,输入一个整数,输出该属的二进制表示中1的个数。例如把9表示为2进制是1001,有2位是1。因此,如果输入9,则该函数输出2。
错误的方法
这是很基本的考察二进制的问题,很容易形成一个思路,把输入的数与1与运算(&),然后右移,每次都与1与运算。结果是1就把计数加1。直到输入的数变成0。代码如下:
public int hammingWeight(int n) {
int count = 0;
while (n != 0) {
if ((n & 1) != 0) {
count++;
}
n = n >> 1;
}
return count;
}
这个算法是错误的,原因是因为右移符号并不是简单的右移动,他会根据这个数的符号在前面补数,是负数的话就会补1,是正数补0。所以这个方法对正数是合适的,对负数的话会形成死循环。
常规解法
为了避免死循环,我们可以不右移数字n(输入的数字)。首先先把n和1坐与运算,判断n的最低位是不是1。接着把1左移一位得到2,再和n做与运算,就能判断n的次低位是不是1.基于这种思路,代码可以这样写:
public int hammingWeight(int n) {
int count = 0;
int m = 1;
while (m != 0) {
if ((n & m) != 0) {
count++;
}
m <<= 1;
}
return count;
}
这个解法中,循环的次数等于整数二进制的位数,32位的整数循环32此。下面这个算法,整数中有几个1就只需要循环几次。
看完说“卧槽”的解法
分析这种算法之前,我们先来分析一下把一个数减去1的情况。如果一个整数不等于0,那么该整数的二进制中至少有一位是1。
- 情况一:最右边是1,最右边是1的话,这个整数减1,只相当于把最后一位从1变成0,
前1010011001
后1010011000
- 情况二:最右边不是1,最右边不是1的话,假设最右边的1在m位,减1操作会将m位变成0,m位右边(右边只有0)的全部变成1。
前1010011000
后1010010111
我们综合两种情况用n&(n-1),得到的结果其实是将n的最右边1的位变成0。
我们把上面的结论总结起来就是:把一个整数减去1,在和原整数做与运算,会把该整数最右边的1变成0。那么一个整数二进制有多少1,就可以进行多少次这样的操作。
public int hammingWeight(int n) {
int count = 0;
while (n != 0) {
++count;
n = n & (n - 1);
}
return count;
}
咋样,是不是感觉神奇的很,很多问题,用我们常规思路不好解决,但是用位运算可能简单到发指。