Create: 338-Counting-Bits.c

nick322 · Sep 4, 2022 · 8652eff · 8652eff
1 parent 748742c
commit 8652eff
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/c/338-Counting-Bits.c b/c/338-Counting-Bits.c
@@ -0,0 +1,20 @@
+/**
+ * Note: The returned array must be malloced, assume caller calls free().
+ */
+int* countBits(int n, int* returnSize){
+ // Initialize array of size n + 1
+ int arraySize = n + 1;
+ *returnSize = arraySize;
+ int *dp = (int *)malloc(sizeof(int)*arraySize);
+ memset(dp, 0, arraySize*sizeof(dp[0]));
+ int offset = 1;
+
+ // Perform dp
+ for (int i = 1; i <= n; i++) {
+ if (offset * 2 == i) {
+ offset = i;
+ }
+ dp[i] = 1 + dp[i - offset];
+ }
+ return dp;
+}