Time: 20190904
Type: Medium
题目描述
给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。
示例 1:
输入: 2
输出: [0,1,1]
示例 2:
输入: 5
输出: [0,1,1,2,1,2]
进阶:
给出时间复杂度为O(n*sizeof(integer))的解答非常容易。但你可以在线性时间O(n)内用一趟扫描做到吗?
要求算法的空间复杂度为O(n)。
你能进一步完善解法吗?要求在C++或任何其他语言中不使用任何内置函数(如 C++ 中的 __builtin_popcount)来执行此操作。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/counting-bits
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
用高层语言Python提供的特性可以直接暴力求解:
class Solution:
def countBits(self, num: int) -> List[int]:
def getBits(num):
s = bin(num)
return s.count('1')
res = []
for i in range(num + 1):
res += [getBits(i)]
return res
其中,bin()
函数可以得到整数的二进制原码表示,比如:
- bin(3) ⇒ ‘0b11’
- bin(-12) => ‘-0b1100’
然后再统计字符串中的1
的个数,这是非常冗余的求法,算是对题目有进一步的认知。
显然这不是我们要的算法。
动态规划算法
本题的动态规划法是比较巧妙且易理解的。
f [ n ] = f [ n / / 2 ] + 1 , 如 果 n 是 奇 数 f [ n ] = f [ n / / 2 ] , 如 果 n 是 偶 数 f[n] = f[n // 2] + 1, 如果n是奇数 \\ f[n] = f[n // 2], 如果n是偶数 f[n]=f[n//2]+1,如果n是奇数f[n]=f[n//2],如果n是偶数
这里,状态f[n]
表示数字n
的原码二进制中1的个数。
奇数最后一位二进制表示一定是1,偶数一定是0,除以2表示右移,所以奇数会丢一个1
,偶数不会。
递推公式出来,再补上基本条件,问题得解。
代码
class Solution:
def countBits(self, num: int) -> List[int]:
f = [0 for _ in range(num + 1)]
if num > 0:
f[0] = 0
f[1] = 1
for i in range(2, num + 1):
if i % 2 == 0:
f[i] = f[i // 2]
else:
f[i] = f[i // 2] + 1
return f
END.