aoc/year2024/
day02.rs

1//! # Red-Nosed Reports
2//!
3//! Computes boths parts simultaneously. Each pair of levels is converted into deltas of
4//! either +1, -1 or 0. For example:
5//!
6//! * `1 3 6 7 9` => `+1 +1 +1 +1`
7//! * `9 7 6 2 1` => `-1 -1 0 -1`
8//!
9//! If the sum of all delta equals ±4, then we know that all levels are increasing or
10//! decreasing. Any other value indicates either mixed up and down transitions or levels that
11//! are too far apart.
12//!
13//! For part two we remove each pair of deltas (or single delta at each end) then replace with
14//! the sum of the delta from the new neighbors on either side.
15use crate::util::parse::*;
16
17type Input = (u32, u32);
18
19/// Minimize allocation to only a single `vec` reused for each report.
20pub fn parse(input: &str) -> Input {
21    let mut report = Vec::new();
22
23    input.lines().fold((0, 0), |(part_one, part_two), line| {
24        report.clear();
25        report.extend(line.iter_signed::<i32>());
26
27        let (p1, p2) = check(&report);
28        (part_one + p1, part_two + p2)
29    })
30}
31
32pub fn part1(input: &Input) -> u32 {
33    input.0
34}
35
36pub fn part2(input: &Input) -> u32 {
37    input.1
38}
39
40fn check(report: &[i32]) -> (u32, u32) {
41    let size = report.len();
42    let score: i32 = report.windows(2).map(|w| delta(w[0], w[1])).sum();
43
44    if score.abs() == (size - 1) as i32 {
45        return (1, 1);
46    }
47
48    for i in 0..size {
49        let mut score = score;
50
51        // Snip out each level and replace with new level computed from neighbors to either side.
52        if i > 0 {
53            score -= delta(report[i - 1], report[i]);
54        }
55        if i < size - 1 {
56            score -= delta(report[i], report[i + 1]);
57        }
58        if i > 0 && i < size - 1 {
59            score += delta(report[i - 1], report[i + 1]);
60        }
61
62        if score.abs() == (size - 2) as i32 {
63            return (0, 1);
64        }
65    }
66
67    (0, 0)
68}
69
70/// Convert each pair of levels to either +1 for increase, -1 for decrease or 0 for invalid range.
71fn delta(a: i32, b: i32) -> i32 {
72    let diff = b - a;
73    (diff.abs() <= 3) as i32 * diff.signum()
74}