Add problem 3163: String Compression III

Author: EFanZh

src/lib.rs

@@ -2115,6 +2115,7 @@ pub mod problem_3158_find_the_xor_of_numbers_which_appear_twice;
 pub mod problem_3159_find_occurrences_of_an_element_in_an_array;
 pub mod problem_3160_find_the_number_of_distinct_colors_among_the_balls;
 pub mod problem_3162_find_the_number_of_good_pairs_i;
+pub mod problem_3163_string_compression_iii;
 pub mod problem_3164_find_the_number_of_good_pairs_ii;
 pub mod problem_3350_adjacent_increasing_subarrays_detection_ii;
 

src/problem_3163_string_compression_iii/iterative.rs

@@ -0,0 +1,54 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn compressed_string(word: String) -> String {
+        let mut result = String::new();
+        let mut iter = word.bytes();
+
+        'outer: while let Some(mut first) = iter.next() {
+            let mut count = b'1';
+
+            loop {
+                if let Some(c) = iter.next() {
+                    if c == first {
+                        if count < b'8' {
+                            count += 1;
+                        } else {
+                            result.extend(['9', char::from(first)]);
+
+                            break;
+                        }
+                    } else {
+                        result.extend([char::from(count), char::from(first)]);
+                        first = c;
+                        count = b'1';
+                    }
+                } else {
+                    result.extend([char::from(count), char::from(first)]);
+
+                    break 'outer;
+                }
+            }
+        }
+
+        result
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn compressed_string(word: String) -> String {
+        Self::compressed_string(word)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_3163_string_compression_iii/mod.rs

@@ -0,0 +1,18 @@
+pub mod iterative;
+
+pub trait Solution {
+    fn compressed_string(word: String) -> String;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [("abcde", "1a1b1c1d1e"), ("aaaaaaaaaaaaaabb", "9a5a2b")];
+
+        for (word, expected) in test_cases {
+            assert_eq!(S::compressed_string(word.to_string()), expected);
+        }
+    }
+}