LeetCode每日一题:比特位计数(No.338)

515 阅读1分钟

题目:比特位计数


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

示例:


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

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

思考:


二进制数中,奇数中的1的个数,与比它小1的偶数相比多一个,即个位上的1。
例如: 3  11
      2  10 
      5  101
      4  100
二进制中,偶数中的1的个数,等于原偶数除以2的结果中的1的个数。因为偶数末尾为0,除以2,二进制中即为去掉末尾的0。
例如: 4  100
      2  10
      1  1
又因为数字0中1的个数为0,所以有1中1的个数为1,2中1的个数等于2/2=1中1的个数,以此类推。

实现:


 class Solution {
    public int[] countBits(int num) {
        int[] result = new int[num + 1];
        result[0] = 0;
        for (int count = 1; count <= num; count++) {
            if (count % 2 == 1) {
                result[count] = result[count - 1] + 1;
            } else {
                result[count] = result[count / 2];
            }
        }
        return result;
    }
}