From afd1a1a10aa4baf3b87073174648e8d090a5b035 Mon Sep 17 00:00:00 2001
From: _tud <98935832+UnderscoreTud@users.noreply.github.com>
Date: Mon, 2 Dec 2024 16:25:34 +0300
Subject: [PATCH] Make day 2 cleaner

---
 src/bin/02.rs | 34 +++++++++++++---------------------
 1 file changed, 13 insertions(+), 21 deletions(-)

diff --git a/src/bin/02.rs b/src/bin/02.rs
index 0c2eb86..433bdb4 100644
--- a/src/bin/02.rs
+++ b/src/bin/02.rs
@@ -1,35 +1,27 @@
+use std::cmp::Ordering::Equal;
+use itertools::Itertools;
+
 advent_of_code::solution!(2);
 
 fn is_safe(data: &Vec<u32>) -> bool {
-    let mut previous_direction = None;
-    for i in 0..(data.len() - 1) {
-        let diff = data[i].abs_diff(data[i + 1]);
-        if diff == 0 || diff > 3 {
-            return false;
-        }
-        let direction = data[i].cmp(&data[i + 1]);
-        match previous_direction {
-            Some(previous_direction) => if previous_direction != direction {
-                return false;
-            },
-            None => previous_direction = Some(direction)
-        }
+    let direction = data[0].cmp(&data[1]);
+    if direction == Equal {
+        return false;
     }
-    true
+    data.iter().tuple_windows::<(&u32, &u32)>()
+        .map(|(current, next)| (1..=3).contains(&current.abs_diff(*next)) && current.cmp(next) == direction)
+        .all(|valid| valid)
 }
 
 fn is_safe_lenient(data: &Vec<u32>) -> bool {
     if is_safe(data) {
         return true;
     }
-    for i in 0..data.len() {
+    (0..data.len()).any(|index| {
         let mut modified_data = data.clone();
-        modified_data.remove(i);
-        if is_safe(&modified_data) {
-            return true
-        }
-    }
-    false
+        modified_data.remove(index);
+        is_safe(&modified_data)
+    })
 }
 
 fn count_safe_reports<P>(input: &str, predicate: P) -> u32