【位运算】leetcode 191. 位1的个数

191. 位1的个数

题目描述

编写一个函数，输入是一个无符号整数（以二进制串的形式），返回其二进制表达式中数字位数为 ‘1’ 的个数（也被称为汉明重量）。

提示

• 请注意，在某些语言（如 Java）中，没有无符号整数类型。在这种情况下，输入和输出都将被指定为有符号整数类型，并且不应影响您的实现，因为无论整数是有符号的还是无符号的，其内部的二进制表示形式都是相同的。

• 在 Java 中，编译器使用二进制补码记法来表示有符号整数。因此，在上面的 示例 3 中，输入表示有符号整数 -3。

示例1：

输入： 00000000000000000000000000001011
输出： 3
解释： 输入的二进制串 00000000000000000000000000001011 中，共有三位为 ‘1’。

示例2：

输入： 00000000000000000000000010000000
输出： 1
解释： 输入的二进制串 00000000000000000000000010000000 中，共有一位为 ‘1’。

示例3：

输入： 11111111111111111111111111111101
输出： 31
解释： 输入的二进制串 11111111111111111111111111111101 中，共有 31 位为 ‘1’。

提示

• $输入必须是长度为32的二进制串$ 。

方法一：循环检查二进制位

解题思路

直接循环检查给定整数 n 的二进制位的每一位是否为 1。
把 n 每次和 $2^i$ 相与，就可以判断 n 的第 i 位是否为 1。

代码

class Solution {
public:
    int hammingWeight(uint32_t n) {
        int res = 0;
        for(int i = 0; i < 32; i++)
            if(n & (1 << i))
                res++;
        return res;
    }
};

复杂度分析

• 时间复杂度：O(k)，其中 k 是 $\text{int}$ 型的二进制位数，k=32。我们需要检查 n 的二进制位的每一位，一共需要检查 32 位。
• 空间复杂度：O(1)。

方法二：位运算优化

解题思路

n & (n - 1)，这个操作可以把 n 的二进制的最低位的 1 变成 0。
例如：6(110) & 5(101) = 4(100)。
不断循环，把 n 和 n - 1 相与，直到 n 为 0，循环了几次就说明 n 的二进制有多少个 1。

代码

class Solution {
public:
    int hammingWeight(uint32_t n) {
        int res = 0;
        while(n)
        {
            n = n & (n - 1);
            res++;
        }
        return res;
    }
};

复杂度分析

• 时间复杂度：$O(logn)$。循环次数等于 n 的二进制位中 1 的个数，最坏情况下 n 的二进制位全部为 1。我们需要循环 $\log n$ 次。
• 空间复杂度：$O(1)$。

方法三：lowbit 解法

解题思路

n & -n，这个操作可以把 n 的二进制变成只剩最后一位 1。
然后每循环一次，n 减去 lowbit(n)，n 的二进制的最后 1 位 1 变成 0。循环几次就说明 n 的二进制有几个 1。

代码

class Solution {
public:
    int lowbit(uint32_t n) {
        return n & -n;
    }
    int hammingWeight(uint32_t n) {
        int res = 0;
        while(n)
        {
            n -= lowbit(n);
            res++;
        }
        return res;
    }
};

复杂度分析

• 时间复杂度：$O(logn)$。循环次数等于 n 的二进制位中 1 的个数，最坏情况下 n 的二进制位全部为 1。我们需要循环 $\log n$ 次。
• 空间复杂度：$O(1)$。