leetcode-338. counting bits

public int[] countBits(int num) {
    int[] count = new int[num + 1];
    count[0] = 0;
    for(int i = 1; i <= num; i++) {
        count[i] = count[i/2] + i % 2;
    }
    return count;
}