338

class Solution:
    def countBits(self, num: int) -> List[int]:

        dp = [0] * (num + 1)
        dp[0] = 0

        diff = 1
        steps = 1
        for i in range(1, num+1):
            if steps:
                steps -= 1
            else:
                diff *= 2
                steps = diff - 1
            dp[i] = dp[i-diff] + 1

        return dp