Initial commitHEADmain

1 files changed, 23 insertions, 0 deletions

diff --git a/counting-bits/src/main.rs b/counting-bits/src/main.rs
new file mode 100644
index 0000000..1203594
--- /dev/null
+++ b/counting-bits/src/main.rs
@@ -0,0 +1,23 @@
+fn main() {
+    let tests = [
+        (2, vec![0,1,1]),
+        (5, vec![0,1,1,2,1,2])
+    ];
+
+    for test in tests {
+        println!("{} is {:?} should be {:?}", test.0, count_bits(test.0), test.1);
+    }
+}
+
+fn count_bits(n: i32) -> Vec<i32> {
+    let mut solution: Vec<i32> = Vec::new();
+    for i in 0..=n {
+        let mut ones = 0;
+        for c in 0..32 {
+            ones += (1<<c & i != 0) as i32;
+        }
+        solution.push(ones);
+    }
+    
+    return solution;
+}