338. 比特位计数（leetcode）

338. 比特位计数（leetcode）

题目描述

给你一个整数 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$

题解1（C++版本）

class Solution {
public:
    vector<int> countBits(int n) {
        vector<int> ans(n + 1, 0);
        for(int i = 1; i <= n; i++){
            for(int j = 0; j <= 20; j++){
                if(i & (1<<j)) ans[i]++;
            }
        }
        return ans;
    }
};

题解2（C++版本）

class Solution {
public:
    vector<int> countBits(int n) {
        vector<int> dp(n + 1, 0); // 对于一个二进制数来说，如果它的最低位为1,则它与 n/2 的 1 的个数相差1。 如果它的最低位为 0，则它与 n/2 的 1 的个数相同
        for(int i = 1; i <= n; i++){
            if(i & 1) dp[i] = dp[i/2] + 1;
            else dp[i] = dp[i/2];
        }
        return dp;
    }
};