剑指 Offer II 003. 前 n 个数字二进制中1的个数

题目链接

思路1

• 整数$num$每进行一次$num\&(num-1)$运算，$num$的二进制数就减少一个最右边的$1$，然后对$num$进行赋值$num=num\&(num-1)$，直到$num=0$，循环的次数即是$num$中$1$的个数

代码1

class Solution {
    public int[] countBits(int n) {
        int[] ans = new int[n+1]; //隐式初始化，数组全为0
        for(int i = 1; i <= n; i++) {
            int num = i;
            for(; ;) {
                if(num == 0) break;
                ans[i]++;
                num = num&(num-1);
            }
        }
        return ans;
    
    }
}

思路2

• 整数$num$二进制中$1$的个数即是整数$num\&(num-1)$二进制中$1$的个数$+1$

代码2

class Solution {
    public int[] countBits(int n) {
        int[] ans = new int[n+1];
        for(int i = 1; i <= n; i++) {
            ans[i] = ans[i&(i-1)] + 1;
        }
        return ans;
    }
}

思路3

• 整数$num$若是偶数，它的二进制中1的个数与$num/2$相同；整数$num$若是奇数，它的二进制中1的个数与$num/2$二进制中1的个数+1；
• 这里用$i>>1$表示$i/2$，因为位运算比除法运算和求余运算更高效。

代码3

class Solution {
    public int[] countBits(int n) {
        int[] ans = new int[n+1];
        for(int i = 1; i <= n; i++) {
            //System.out.println("ans[i>>1]= "+ans[i>>1]+" i&1= "+(i&1));
            ans[i] = ans[i>>1] + (i&1);//i&1一定要加括号
        }
        return ans;
    }
}