LeetCode解题笔记 44 —— 338. 比特位计数

题目

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。

示例 1:

输入: 2
输出: [0,1,1]

示例 2:

输入: 5
输出: [0,1,1,2,1,2]

 

解法

class Solution {
    public int[] countBits(int num) {
        int[] res = new int[num+1];
        res[0] = 0;
        if(num>0){
            res[1] = 1;
        }
        int index = 2;
        int z = 2;
        while(z <= num){
            int temp = index*2;
            for(;index < temp && index <=num; index++){
                //2~3是0~1的基础上加上最高位的一个1，4~7是0~3的基础上加上最高位的一个1，如此类推，每次周期*2
                res[index] = 1 + res[index - z];
            }
            z = temp;
        }
        return res;
    }
}