LeetCode 338.比特位计数

题目描述

给你一个整数 $n$ ，对于 $0<=i<=n$ 中的每个 $i$ ，计算其二进制表示中 $1$ 的个数 ，返回一个长度为 $n+1$ 的数组 $ans$ 作为答案。

示例 1：

输入：n = 2
输出：[0,1,1]
解释：
0 --> 0
1 --> 1
2 --> 10

示例 2：

输入：n = 5
输出：[0,1,1,2,1,2]
解释：
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101

提示：

• $0<=n<=10^5$

进阶：

• 很容易就能实现时间复杂度为 $O(nlogn)$ 的解决方案，你可以在线性时间复杂度 $O(n)$ 内用一趟扫描解决此问题吗？
• 你能不使用任何内置函数解决此问题吗？（如，C++ 中的 KaTeX parse error: Expected group after '_' at position 1: _̲_builtin_popcou… ）

---

思路

如果一个数x是偶数，那么x的二进制的1的个数 = x / 2的二进制的1的个数
如果一个数x是奇数，那么x的二进制的1的个数 = x / 2的二进制的1的个数 + 1
从0到n遍历求就可以了，类似于动态规划的遍历。

C++ 代码

class Solution {
public:
    vector<int> countBits(int n) {
        vector<int> ret(n + 1);
        // 第一个数是0
        ret[0] = 0;
        // 如果n不为0 n大于0 进行后面的运算
        if (n != 0) ret[1] = 1;
        // 遍历
        for (int i = 2; i <= n; i++)
        {
            if (i % 2 == 0) ret[i] = ret[i / 2];
            else ret[i] = ret[i / 2] + 1;
        }

        return ret;
    }
};