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()));
}