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++版本)
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;
}
};