leetcode 338. 比特位计数 python

给你一个整数 n ，对于 0 <= i <= n 中的每个 i ，计算其二进制表示中 1 的个数 ，返回一个长度为 n + 1 的数组 ans 作为答案。

示例 1：

输入：n = 2
输出：[0,1,1]
解释：
0 --> 0
1 --> 1
2 --> 10

示例 2：

输入：n = 5
输出：[0,1,1,2,1,2]
解释：
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101

提示：

• 0 <= n <= 105

  class Solution:
      def countBits(self, n: int) -> List[int]:
          ans = []
          for i in range(n+1):
              x = i
              cnt = 0
              while(x!=0):
                  if(x%2!=0):
                      cnt+=1
                  x //= 2
              ans.append(cnt)
          return ans