Leetcode 338.比特位计数

发表于 | 更新于 | 分类于

题目描述

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

示例 1:

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

示例 2:

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

题目分析

假如某一个数的二进制1位数是n, 则它左移一位的数的1位数也是n, 这样我们便可以使用筛法来避免重复运算.

源码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
int finded[100000] = {0};

class Solution {
public:
    int dp(int num, int max) {
        if(finded[num] == 0) {
            int t = num, sum = 0;
            while(t != 0) {
                sum += t & 1;
                t>>=1;
            }
            t = num;
            while(t <= max) {
                finded[t] = sum;
                t <<= 1;
            }
        }
        return finded[num];
    }

    vector<int> countBits(int num) {
        vector<int> res;
        res.push_back(0);
        for(int i=1; i<=num; ++i) {
            res.push_back(dp(i, num));
        }
        return res;
    }
};
0%