338. 比特位计数（leetcode）

lulutoy2024-06-30 13:33

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 < = 1 0 5 0 <= n <= 10^5 0<=n<=105

题解1（C++版本）

cpp 复制代码

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++版本）

cpp 复制代码

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;
    }
};