public int[] countBits(int n) {
    int result[] = new int[n + 1];
    for (int i = 0; i <= n; i++) {
        int num = i;
        int count = 0;
        while (num > 0) {
            count += num & 1;
            num = num >> 1;
        }
        result[i] = count;
    }
    return result;
}