Update 338_Counting_Bits.java

Author: Sean Prashad

Bitwise Operations/338_Counting_Bits.java

@@ -1,16 +1,15 @@
 class Solution {
     public int[] countBits(int num) {
-        int[] dp = new int[num + 1];
-        int offset = 1;
+        int[] result = new int[num + 1];
 
         for (int i = 1; i <= num; i++) {
-            if (offset * 2 == i) {
-                offset *= 2;
+            if (i % 2 == 0) {
+                result[i] = result[i >> 1];
+            } else {
+                result[i] = result[i - 1] + 1;
             }
-
-            dp[i] = dp[i - offset] + 1;
         }
 
-        return dp;
+        return result;
     }
 }