【中等】338-比特位计数 Counting Bits

题目

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。

输入: 2
输出: [0,1,1]
输入: 5
输出: [0,1,1,2,1,2]

解法

解法1

我们将数字按照位数进行分组

1位:
0(0)
1(1)

2位:
10(2)
11(3)

3位:
100(4)
101(5)
110(6)
111(7)
...

可以发现,2位的数字如果把最高位的1去掉之后可以与1位的数字一一对应,也就是说数字2/3的个数是数字0/1的个数+1

3位的数字如果把最高位的1去掉之后可以与1位和2位的数字一一对应,也就是说数字4/5/6/7的个数是数字0/1/2/3的个数+1

此时只需要按位数将对应的数字压入vector,然后从头遍历+1即可

代码

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> res;
        res.push_back(0);
        while(num > 0){
            int size = res.size();
            for(int i = 0; i < size && num>0; ++i){
                res.push_back(res[i]+1);
                --num;
            }
        }
        return res;
    }
};

解法2

在leetcode题解的最短用时解法中,代码用了__builtin_popcount()这样一个内置函数,这个函数是专门用来计算bit位数字的,属于一个取巧解法,不推荐

posted @ 2021-03-03 09:22  陌良  阅读(59)  评论(0编辑  收藏  举报