summaryrefslogtreecommitdiff
path: root/counting-bits/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'counting-bits/src/main.rs')
-rw-r--r--counting-bits/src/main.rs23
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;
+}