67: Submit

sankichi92 · Dec 31, 2023 · 7bb9204 · 7bb9204
1 parent 59e6bf4
commit 7bb9204
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/67-counting-bits/src/lib.rs b/67-counting-bits/src/lib.rs
@@ -1,14 +1,20 @@
-pub fn add(left: usize, right: usize) -> usize {
-    left + right
+pub fn count_bits(n: i32) -> Vec<i32> {
+    (0..=n)
+        .map(|i| format!("{i:b}").chars().filter(|&c| c == '1').count() as i32)
+        .collect()
 }
 
 #[cfg(test)]
 mod tests {
     use super::*;
 
     #[test]
-    fn it_works() {
-        let result = add(2, 2);
-        assert_eq!(result, 4);
+    fn case1() {
+        assert_eq!(count_bits(2), vec![0, 1, 1]);
+    }
+
+    #[test]
+    fn case2() {
+        assert_eq!(count_bits(5), vec![0, 1, 1, 2, 1, 2]);
     }
 }