day 2 part 2

src/day02.rs

@@ -1,14 +1,26 @@
-use std::fs::read_to_string;
+use std::{default, fs::read_to_string};
 
 use itertools::Itertools;
 
 pub fn day_main() {
     let input = read_to_string("input/day02.txt").unwrap();
     println!("part1: {}", part1(&input));
-    // println!("part2: {}", part2(&input));
+    println!("part2: {}", part2(&input));
 }
 
 fn part1(input: &str) -> usize {
+    input
+        .lines()
+        .map(|line| {
+            line.split_whitespace()
+                .map(|s| s.parse::<i32>().unwrap())
+                .collect_vec()
+        })
+        .filter(|report: &Vec<i32>| is_safe(report))
+        .count()
+}
+
+fn part2(input: &str) -> usize {
     input
         .lines()
         .map(|line| {
@@ -17,6 +29,19 @@ fn part1(input: &str) -> usize {
                 .collect_vec()
         })
         .filter(|report: &Vec<i32>| {
+            (0..report.len()).any(|del| {
+                let without = report
+                    .iter()
+                    .enumerate()
+                    .filter_map(|(i, e)| if del != i { Some(*e) } else { None })
+                    .collect_vec();
+                is_safe(&without)
+            })
+        })
+        .count()
+}
+
+fn is_safe(report: &[i32]) -> bool {
     let pairs = report
         .windows(2)
         .map(|window| window[1] - window[0])
@@ -24,11 +49,8 @@ fn part1(input: &str) -> usize {
     let steepest_slope = pairs.iter().map(|x| x.abs()).max().unwrap();
     let lowest_slope = pairs.iter().map(|x| x.abs()).min().unwrap();
     let same_sign = pairs.iter().map(|x| x.signum()).unique().count() == 1;
-            println!("{:?} - {steepest_slope}/{same_sign}", report);
 
     steepest_slope <= 3 && lowest_slope >= 1 && same_sign
-        })
-        .count()
 }
 
 // fn part2(input: &str) -> Int {