Advent-of-Code/2024/day-02/src/part2.rs

use miette::Result;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum Level {
    Safe,
    Unsafe,
}

impl From<&Vec<usize>> for Level {
    fn from(value: &Vec<usize>) -> Self {
        if is_safe(value) {
            return Self::Safe;
        }
        for (idx, _) in value.iter().enumerate() {
            let mut new_levels = value.clone();
            new_levels.remove(idx);
            if is_safe(&new_levels) {
                return Self::Safe;
            }
        }
        Self::Unsafe
    }
}

fn is_safe(levels: &[usize]) -> bool {
    (levels.iter().is_sorted() || levels.iter().rev().is_sorted())
        && levels.windows(2).all(|pairs| {
            let diff = pairs[0].abs_diff(pairs[1]);
            (1..=3).contains(&diff)
        })
}

#[tracing::instrument]
pub fn process(input: &str) -> Result<usize> {
    let sum = input
        .lines()
        .map(|line| {
            let levels = line
                .split_whitespace()
                .map(|x| x.parse::<usize>().unwrap())
                .collect::<Vec<_>>();

            Level::from(&levels)
        })
        .filter(|&report| report == Level::Safe)
        .count();
    Ok(sum)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_process() -> Result<()> {
        let input = "7 6 4 2 1
1 2 7 8 9
9 7 6 2 1
1 3 2 4 5
8 6 4 4 1
1 3 6 7 9
";
        let result = 4;
        assert_eq!(process(input)?, result);
        Ok(())
    }
}