Create: 338-Counting-Bits.go

Gyanachand1 · Sep 14, 2022 · b26f565 · b26f565
1 parent 9e82845
commit b26f565
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/go/338-Counting-Bits.go b/go/338-Counting-Bits.go
@@ -0,0 +1,12 @@
+func countBits(n int) []int {
+ dp := make([]int, n + 1)
+ offset := 1
+
+ for i := 1; i <= n; i++ {
+ if offset * 2 == i {
+ offset = i
+ }
+ dp[i] = 1 + dp[i - offset]
+ }
+ return dp
+}