[rust] Day 5.1
change
commit 224ebd6de7e63fc3ac5ed61d9080f1b7385b5e7c
author Alpha Chen <alpha.chen@gmail.com>
date
parent 3aa6c39c
diff --git a/rust/src/day_05.rs b/rust/src/day_05.rs
index c98afc4..edec83f 100644
--- a/rust/src/day_05.rs
+++ b/rust/src/day_05.rs
@@ -8,11 +8,11 @@ pub struct Day05 {
     input: String,
 }
 
+#[allow(dead_code)]
 impl Day05 {
     fn is_nice(string: &String) -> bool {
-        Day05::has_three_vowels(string) &&
-            Day05::has_double_letters(string) &&
-            Day05::has_no_substrings(string)
+        Day05::has_pair_of_two_letters(string) &&
+            Day05::has_letter_sandwich(string)
     }
 
     fn has_three_vowels(string: &String) -> bool {
@@ -27,6 +27,17 @@ impl Day05 {
     fn has_no_substrings(string: &String) -> bool {
         !vec!["ab", "cd", "pq", "xy"].iter().any(|&s| string.contains(s))
     }
+
+    fn has_pair_of_two_letters(string: &String) -> bool {
+        string.as_bytes().windows(2).any(|win| {
+            let s = String::from_utf8(win.to_vec()).unwrap();
+            string.split(&s).count() > 2
+        })
+    }
+
+    fn has_letter_sandwich(string: &String) -> bool {
+        string.as_bytes().windows(3).any(|win| win[0] == win[2])
+    }
 }
 
 impl Day for Day05 {
@@ -41,9 +52,8 @@ impl Day for Day05 {
 
 #[test]
 fn test_nice() {
-    assert!( Day05::is_nice(&"ugknbfddgicrmopn".to_string()));
-    assert!( Day05::is_nice(&"aaa".to_string()));
-    assert!(!Day05::is_nice(&"jchzalrnumimnmhp".to_string()));
-    assert!(!Day05::is_nice(&"haegwjzuvuyypxyu".to_string()));
-    assert!(!Day05::is_nice(&"dvszwmarrgswjxmb".to_string()));
+    assert!( Day05::is_nice(&"qjhvhtzxzqqjkmpb".to_string()));
+    assert!( Day05::is_nice(&"xxyxx ".to_string()));
+    assert!(!Day05::is_nice(&"uurcxstgmygtbstg ".to_string()));
+    assert!(!Day05::is_nice(&"ieodomkazucvgmuy ".to_string()));
 }