leetcode-338:比特位计数

一枚小白

题目描述

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

示例 1:

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

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

思路

假设i的二进制中1的位数有f(i),则f(i>>1)1的位数有2种情况:

$$ f(i>>1)= \begin{cases} f(i) & \text{i的最后一位是0}\\ f(i)-1 & \text{i的最后一位是1} \end{cases} $$

已知 i>>1 小于i,所以f(i>>1)会先被计算出来,所以f(i)的公式如下:

$$ f(i)= \begin{cases} f(i>>1) & \text{i的最后一位是0} \\ f(i>>1)+1 & \text{i的最后一位是1} \end{cases} $$

代码

public int[] countBits(int num) {
    int[] bitsCount = new int[num + 1];
    for (int i = 1; i <= num; i++) {
        bitsCount[i] = bitsCount[i >> 1] + (1 & i);
    }
    return bitsCount;
}
个人博客中本文链接,欢迎查看:天天説
阅读 1.2k

程序猿GG的日常
日常生活随手记,是与其他人交流沟通知识的平台。如果您有想了解的内容,请告知我进行补充,或者您给本...

一枚小白

244 声望
3 粉丝
0 条评论

一枚小白

244 声望
3 粉丝
文章目录
宣传栏