feat: easy: divide_a_string_into_groups_of_size_k.rs

Author: illiafox

src/easy/divide_a_string_into_groups_of_size_k.rs

@@ -0,0 +1,36 @@
+struct Solution;
+
+impl Solution {
+    pub fn divide_string(s: String, k: i32, fill: char) -> Vec<String> {
+        let mut out = vec![];
+        let k = k as usize;
+
+        let mut i = 0;
+        while i < s.len() {
+            let mut chunk = s[i..s.len().min(i + k)].to_string();
+            while chunk.len() < k {
+                chunk.push(fill);
+            }
+            out.push(chunk);
+            i += k;
+        }
+
+        out
+    }
+}
+
+#[test]
+fn test() {
+    let test_cases = [
+        ("abcdefghi", 3, 'x', vec!["abc", "def", "ghi"]),
+        ("abcdefghij", 3, 'x', vec!["abc", "def", "ghi", "jxx"]),
+    ];
+
+    for (s, k, fill, expected) in test_cases {
+        assert_eq!(
+            Solution::divide_string(s.to_string(), k, fill),
+            expected,
+            "failed for {s}, k {k}, fill {fill}",
+        );
+    }
+}

src/easy/mod.rs

@@ -3,6 +3,7 @@ mod check_if_array_is_sorted_and_rotated;
 mod check_if_one_string_swap_can_make_strings_equal;
 mod clear_digits;
 mod count_pairs_whose_sum_is_less_than_target;
+mod divide_a_string_into_groups_of_size_k;
 mod find_the_difference;
 mod find_words_containing_character;
 mod guess_number_higher_or_lower;