338. Counting Bits

Posted 2020-08-25 三人木君

https://leetcode.com/problems/counting-bits/

给定一个非负数n，输出[0,n]区间内所有数的二进制形式中含1的个数

Example:
For num = 5 you should return [0,1,1,2,1,2].

 

注意fellow up部分，题目说了你要是一个个无脑去遍历输出是不ok的，直接用某些内置函数也是不行的

 

解题思路

实在没思路就看看hint部分

找张纸，多写几个数，包括：

1、数（十进制）

2、数（二进制）

3、二进制中1的个数

 

图片来自http://blog.csdn.net/qiexingqieying/article/details/51719997

感谢作者keke he，侵删

 

横线分组，会发现第1组的count是第0组的count+1，第n组是1+2+。。。+(n-1)组的count+1

所以某个数的count就是这个数减所在组2开方数的count+1

res[x]=res[x-pow(2,len(bin(x))-3)]+1

注意python的bin（）结果强制带前缀0b

然后类似斐波拉切数列，第一个和第二个强制制定

class Solution(object):
    def countBits(self, num):
        res=[1] * (num+1)
        if num==0:
            res[0]=0
        elif num==1:
            res[0]=0
            res[1]=1
        elif num>1:
            res[0] = 0
            res[1] = 1
            for x in xrange(2,num+1):
                res[x]=res[x-pow(2,len(bin(x))-3)]+1
        return res