码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] Counting Bits

时间:2018-01-18 15:01:37      阅读:118      评论:0      收藏:0      [点我收藏+]

标签:numbers   col   run   pos   can   question   cal   proc   problem   

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.

遍历0~num中每个数字,求每个数字二进制中1的个数,并把结果放入数组中。

方法一:easy way。

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> res;
        for (int i = 0; i <= num; i++) 
            res.push_back(countBit(i));
        return res;
    }
    int countBit(int num) {
        int res = 0;
        while (num) {
            if (num & 1)
                res++;
            num = num >> 1;
        }
        return res;
    }
};
// 68 ms

方法二:optimization method (DP)

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> res(num + 1, 0);
        int offset = 1;
        for (int i = 1; i <= num; i++) {
            if (offset * 2 == i)
                offset *= 2;
            res[i] = res[i - offset] + 1;
        }
        return res;
    }
};
// 78 ms

该方法具体解法详见:How we handle this question on interview [Thinking process + DP solution]

 

[LeetCode] Counting Bits

标签:numbers   col   run   pos   can   question   cal   proc   problem   

原文地址:https://www.cnblogs.com/immjc/p/8309923.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!