【leetcode】338. Counting Bits(C++ & Python)

来源:互联网 发布:k最短路径算法 matlab 编辑:程序博客网 时间:2024/05/21 23:52

338. Counting Bits

题目链接

338.1 题目描述:

Given a non negative integer number num. For every numbers i in the range 0 ≤ i ≤ num calculate the number of 1’s in their binary representation and return them as an array.

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

Follow up:

It is very easy to come up with a solution with run time O(n*sizeof(integer)). But can you do it in linear time O(n) /possibly in a single pass?
Space complexity should be O(n).
Can you do it like a boss? Do it without using any builtin function like __builtin_popcount in c++ or in any other language.

338.2 解题思路:

  1. 思路一:分为两步,一步计算每个数字的二进制中1的个数,一步将num内所有数字的二进制中1的个数放入vector中。

    • 计算二进制中1的个数:根据十进制转二进制定义,如果num大于0,每次都对num取余,如果余数为1,则计数count++。num除以2。遍历结束,返回count。
    • 计算num内所有数字的二进制中1的个数:从1到num,依次计算二进制中1的个数并放入vector中。
  2. 思路二:(没太想明白,参考leetcode-discuss)初始化vector大小为num+1,且初始值全为0。从1开始遍历vector。每次vector[i]的值,都是由vector中,(i-1)与i按位与后的坐标下的值加1。至于为什么要定位到i-1与i按位与后的坐标,不太清楚。

  3. 思路三:初始化vector大小为1,初始值为0。

    0—-0个

    1—-1个

    10—-1个
    11—-2个

    100—-1个
    101—-2个
    110—-2个
    111—-3个

    从这个序列中我们可以看到规律,在每个二进制开头,1比0多了一个1,10和11分别比0,1多了一个1,100,101,110,111分别比00,01,10,11多了一个1。根据这个规律,以2的指数为界限增加1。所以可设置J=0,循环,j小于num,获取此时vector大小为n,遍历i从0开始,如果碰到i大于n或者j大于num,则停止遍历。否则,vector中push进vector[i]+1。

  4. 思路四:初始化vector大小为num+1,且初始值全为0。

    0
    1

    10
    11

    100
    101

    110
    111

    从序列中可以看出,从1开始,10和11都是在1的后面加了0或1,100和101都是在10的后面加了0或1,110和111都是在11的后面加了0或1,表示第i个位置的个数是由第i/2位置的个数与i本身奇偶性有关。即vector[i]=vector[i/2]+i%2。

  5. 思路五:初始化vector大小为num+1,且初始值全为0。思路同思路四,只不过i/2等同于i右移1位,即vector[i]=vector[i>>1]+i%2。

338.3 C++代码:

1、思路一代码(69ms):

class Solution140 {public:    int getbit(int num)    {        int count = 0;        while (num>0)        {                   if (num % 2 == 1)                count++;            num = num / 2;        }        return count;    }    vector<int> countBits(int num) {        vector<int>bits(num + 1,0);        for (int i = 1; i < bits.size();i++)        {            bits[i] = getbit(i);        }        return bits;    }};

2、思路二代码(83ms)

class Solution140_1 {public:    vector<int> countBits(int num) {        vector<int>bits(num + 1, 0);        for (int i = 1; i < bits.size(); i++)        {            bits[i] = bits[(i - 1)&i] + 1;        }        return bits;    }};

3、思路三代码(86ms)

class Solution140_2 {public:    vector<int> countBits(int num) {        vector<int>bits(1, 0);        int j = 1;        while (j<=num)        {            int n = bits.size();            for (int i = 0; i < n && j<=num; i++,j++)            {                bits.push_back(bits[i] + 1);            }        }               return bits;    }};

4、思路四代码(93ms)

class Solution140_3 {public:    vector<int> countBits(int num) {        vector<int>bits(num + 1, 0);        for (int i = 1; i < bits.size(); i++)        {            bits[i] = bits[i / 2] + i % 2;        }        return bits;    }};

5、思路五代码(69ms)

class Solution140_4 {public:    vector<int> countBits(int num) {        vector<int>bits(num + 1, 0);        for (int i = 1; i < bits.size(); i++)        {            bits[i] = bits[i>>1] + i % 2;        }        return bits;    }};

338.4 Python代码:

1、思路一代码(428ms)

class Solution(object):    def countBits(self, num):        """        :type num: int        :rtype: List[int]        """        def getbit(num):            count=0            while num>0:                if num%2==1:                    count+=1                num=num/2            return count        bits=[0]*(num+1)        for i in range(1,num+1):            bits[i]=getbit(i)        return bits

2、思路二代码(195ms)

class Solution1(object):    def countBits(self, num):        """        :type num: int        :rtype: List[int]        """        bits=[0]*(num+1)        for i in range(1,num+1):            bits[i]=bits[(i-1)&i]+1        return bits

3、思路三代码(199ms)

class Solution2(object):    def countBits(self, num):        """        :type num: int        :rtype: List[int]        """        bits=[0]        j=1        while j<=num:            i=0            n=len(bits)            while i<n and j<=num:                bits.append(bits[i]+1)                i+=1                j+=1        return bits

4、思路四代码(215ms)

class Solution3(object):    def countBits(self, num):        """        :type num: int        :rtype: List[int]        """        bits=[0]*(num+1)        for i in range(1,num+1):            bits[i]=bits[i/2]+i%2        return bits

5、思路五代码(245ms)

class Solution4(object):    def countBits(self, num):        """        :type num: int        :rtype: List[int]        """        bits=[0]*(num+1)        for i in range(1,num+1):            bits[i]=bits[i>>1]+i%2        return bits

原创粉丝点击