import(solutions): Add problem 338 .

1 files changed, 26 insertions, 0 deletions

diff --git a/works/solutions/cpp/338.cpp b/works/solutions/cpp/338.cpp
new file mode 100644
index 0000000..3b32da4
--- /dev/null
+++ b/works/solutions/cpp/338.cpp
@@ -0,0 +1,26 @@
+#include <vector>
+
+using std::vector;
+
+class Solution
+{
+public:
+    vector<int> countBits(int num)
+    {
+        vector<int> result(num + 1);
+        result[0] = 0;
+        for (int i = 1; i <= num; i++)
+        {
+            if (i % 2 == 1)
+            {
+                result[i] = result[i - 1] + 1;
+            }
+            else
+            {
+                result[i] = result[i / 2];
+            }
+        }
+
+        return result;
+    }
+};