leetcode
leetcode 301 ~ 350
比特位计数

比特位计数

难度:

标签:

题目描述

给你一个整数 n ,对于 0 <= i <= n 中的每个 i ,计算其二进制表示中 1 的个数 ,返回一个长度为 n + 1 的数组 ans 作为答案。

 

示例 1:

输入:n = 2
输出:[0,1,1]
解释:
0 --> 0
1 --> 1
2 --> 10

示例 2:

输入:n = 5
输出:[0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101

 

提示:

  • 0 <= n <= 105

 

进阶:

  • 很容易就能实现时间复杂度为 O(n log n) 的解决方案,你可以在线性时间复杂度 O(n) 内用一趟扫描解决此问题吗?
  • 你能不使用任何内置函数解决此问题吗?(如,C++ 中的 __builtin_popcount

代码结果

运行时间: 20 ms, 内存: 16.8 MB


// 思路:利用Stream的IntStream.rangeClosed生成从0到n的整数流,然后使用map方法计算每个数二进制中1的个数。
// 使用Integer.bitCount来计算1的个数,并将结果收集到数组中。
import java.util.stream.IntStream;
 
public int[] countBits(int n) {
    return IntStream.rangeClosed(0, n)
                    .map(i -> Integer.bitCount(i))
                    .toArray();
}

解释

方法:

该题解采用了动态规划的思路。利用已知的结果来帮助计算后续的结果。基本思路是利用二进制数的性质:任何数x的二进制1的个数,可以由比它小的某个数的二进制1的个数加一得到。具体地,对于当前的数i,若i是2的幂(即i = 2, 4, 8, ...),则i的二进制表示中只有一个1。否则,可以找到最大的2的幂p,使得p < i,然后i的1的个数就是i - p的1的个数加1。这里利用数组ans存储结果,ans[i]表示数i的二进制中1的个数,通过迭代方式计算每一个i从1到n的1的个数。

时间复杂度:

O(n)

空间复杂度:

O(n)

代码细节讲解

🦆
为什么在计算i的二进制中1的个数时,可以通过查找最大的2的幂p来确定i - p的1的个数加1?
在二进制中,每当我们到达一个2的幂,新的一位1将被添加到最左端,其他位均为0(例如2是10,4是100)。这意味着任何数i的1的个数可以看作是i减去这个最大的2的幂p后的数的1的个数加上一个1(因为p是一个2的幂,所以它的二进制中只有一个1)。例如,i = 14,最接近且小于14的2的幂是8(二进制1000),14的二进制是1110,14 - 8 = 6,二进制是110,6的1的个数是2,所以14的1的个数是3。这种方法利用了已知的较小数的结果来高效地计算较大数的结果。
🦆
动态规划在此题中是如何确保每次计算都是基于已知正确的结果进行的?
在动态规划中,问题的解决策略是自底向上的。对于这个问题,我们从最简单的案例开始,即0的二进制中1的个数是0,然后逐步构建到更高的数字。对于每个新的i,我们利用数组ans中已经计算并存储的结果(即i - p的结果),这保证了每一步计算都是基于前面已经验证的正确结果进行的。这种方式确保了整个计算过程的正确性和效率。
🦆
在题解中提到的`cnt`变量具体是用来实现什么功能?它如何帮助管理2的幂的更新?
在这个题解中,`cnt`变量用于跟踪当前数字i与当前的最大2的幂num之间的关系。每次当`cnt`小于`num`时,说明当前的i还没有达到下一个2的幂,我们可以继续利用当前的最大2的幂来计算1的个数;但一旦`cnt`等于`num`,表示达到了下一个2的幂,我们需要重置`cnt`并将`num`翻倍(左移一位),以此来更新我们使用的最大2的幂。这个机制保证了每当i到达一个新的2的幂,我们都正确地更新了2的幂的值和计数器,从而正确地计算接下来的数字的1的个数。

相关问题

位1的个数

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

 

提示:

  • 请注意,在某些语言(如 Java)中,没有无符号整数类型。在这种情况下,输入和输出都将被指定为有符号整数类型,并且不应影响您的实现,因为无论整数是有符号的还是无符号的,其内部的二进制表示形式都是相同的。
  • 在 Java 中,编译器使用二进制补码记法来表示有符号整数。因此,在 示例 3 中,输入表示有符号整数 -3

 

示例 1:

输入:n = 00000000000000000000000000001011
输出:3
解释:输入的二进制串 00000000000000000000000000001011 中,共有三位为 '1'。

示例 2:

输入:n = 00000000000000000000000010000000
输出:1
解释:输入的二进制串 00000000000000000000000010000000 中,共有一位为 '1'。

示例 3:

输入:n = 11111111111111111111111111111101
输出:31
解释:输入的二进制串 11111111111111111111111111111101 中,共有 31 位为 '1'。

 

提示:

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

 

进阶

  • 如果多次调用这个函数,你将如何优化你的算法?