class Solution:
    def countBits(self, n):

        res = []

        for i in range(n + 1):
            # format(i, 'b') --> convert an integer to a binary string without prefix
            res.append(format(i, 'b').count('1'))

        return res