set bit in range

class Solution {
public:
    vector<int> countBits(int num) {
        if( num == 0)
            return {0};

        vector<int> dp(num+1);
        dp[0] = 0;
        dp[1] = 1;

        for(int i = 2 ; i<=num ; i++){
            if(!(i&1))
            {
               dp[i] = dp[i>>1];
            } else {
                dp[i] = dp[i-1] + 1;
            }
        }
        return dp;
    }
};