力扣【338】比特位计数

题目：

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

示例 1:

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

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

题解：就是计算二进制中1的个数

import java.util.*;
class Solution {
    public int[] countBits(int num) {
        int[] res = new int[num + 1];
        for (int i = 0; i < num + 1; i++) {
            int temp = i;
            while (i != 0) {
                res[temp]++;
                i = i & (i - 1);
            }
            i = temp;
        }
        return res;
    }
}

public class Main {
    public static void main(String[] args) {
        int nums = 5;
        Solution p = new Solution();
        int[] res = p.countBits(nums);
        System.out.println("输出结果"+ Arrays.toString(res));
    }
}