Leetcode刷题(第338题)——比特位计数

一、题目

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

二、示例

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

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

三、思路
本题中，我们应该使用位运算，假设当前的数字为i,此时我们可以知道i的位运算结果应该等于i和i-1的AND位运算。并且第一个结果为0。
四、代码展示

/**
 * @param {number} n
 * @return {number[]}
 */
var countBits = function(n) {
    let newArr = new Array(n + 1).fill(0)
    for(let i = 1; i <= n; i++) {
        newArr[i] = newArr[i&(i-1)] + 1
    }
    return newArr
};

五、总结