Create 338-Counting-Bits.py

stenzr · Dec 26, 2021 · b7f9333 · b7f9333
1 parent e46ad16
commit b7f9333
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/338-Counting-Bits.py b/338-Counting-Bits.py
@@ -0,0 +1,10 @@
+class Solution:
+ def countBits(self, n: int) -> List[int]:
+ dp = [0] * (n + 1)
+ offset = 1
+
+ for i in range(1, n + 1):
+ if offset * 2 == i:
+ offset = i
+ dp[i] = 1 + dp[i - offset]
+ return dp