Merge pull request #350 from paulcacheux/day6-2024

Day6 2024

Author: paulcacheux

inputs/2024/day6_test.txt

@@ -0,0 +1,10 @@
+....#.....
+.........#
+..........
+..#.......
+.......#..
+..........
+.#..^.....
+........#.
+#.........
+......#...

results/2024.json

@@ -23,5 +23,10 @@
         "day": 5,
         "part1": "6041",
         "part2": "4884"
+    },
+    {
+        "day": 5,
+        "part1": "4665",
+        "part2": "1688"
     }
 ]

src/aoc2024/day6.rs

@@ -0,0 +1,127 @@
+use crate::aoc2024::Aoc2024;
+use crate::grid::Grid;
+use crate::traits::days::Day6;
+use crate::traits::ParseInput;
+use crate::traits::Solution;
+
+impl ParseInput<Day6> for Aoc2024 {
+    type Parsed = (Grid<bool>, usize, usize);
+
+    fn parse_input(input: &str) -> Self::Parsed {
+        let mut sx = 0;
+        let mut sy = 0;
+        let grid = Grid::parse(input, |c| c);
+
+        let mut bgrid = Grid::new(grid.width, grid.height, false);
+        grid.iter().for_each(|(x, y, &c)| {
+            let val = match c {
+                '#' => true,
+                '.' => false,
+                '^' => {
+                    sx = x;
+                    sy = y;
+                    false
+                }
+                _ => unreachable!(),
+            };
+            bgrid.set(x, y, val);
+        });
+        (bgrid, sx, sy)
+    }
+}
+
+impl Solution<Day6> for Aoc2024 {
+    type Part1Output = usize;
+    type Part2Output = usize;
+
+    fn part1((grid, sx, sy): &(Grid<bool>, usize, usize)) -> usize {
+        compute_path_size(grid, *sx, *sy).unwrap()
+    }
+
+    fn part2((grid, sx, sy): &(Grid<bool>, usize, usize)) -> usize {
+        let mut res = 0;
+        let mut ngrid = grid.clone();
+        for y in 0..grid.height {
+            for x in 0..grid.width {
+                if (x, y) == (*sx, *sy) {
+                    continue;
+                }
+
+                if *grid.get(x, y) {
+                    continue;
+                }
+
+                ngrid.set(x, y, true);
+                if compute_path_size(&ngrid, *sx, *sy).is_none() {
+                    res += 1;
+                }
+                ngrid.set(x, y, false);
+            }
+        }
+
+        res
+    }
+}
+
+const DIRECTIONS: [(isize, isize); 4] = [(0, -1), (1, 0), (0, 1), (-1, 0)];
+
+fn compute_path_size(grid: &Grid<bool>, sx: usize, sy: usize) -> Option<usize> {
+    let (mut x, mut y) = (sx, sy);
+    let mut didx = 0;
+    let mut positions = Grid::new(grid.width, grid.height, [false; 4]);
+
+    loop {
+        let pos = positions.get_mut(x, y);
+        if pos[didx] {
+            return None;
+        }
+        pos[didx] = true;
+
+        let (dx, dy) = DIRECTIONS[didx];
+        if let Some((nx, ny)) = offset_pair(grid, x, y, dx, dy) {
+            if *grid.get(nx, ny) {
+                didx = (didx + 1) % 4;
+            } else {
+                x = nx;
+                y = ny;
+            }
+        } else {
+            break;
+        }
+    }
+
+    Some(
+        positions
+            .iter()
+            .filter(|(_, _, &v)| v.into_iter().any(|b| b))
+            .count(),
+    )
+}
+
+#[inline]
+fn offset_pair(
+    grid: &Grid<bool>,
+    x: usize,
+    y: usize,
+    dx: isize,
+    dy: isize,
+) -> Option<(usize, usize)> {
+    let nx = offset(x, dx, grid.width);
+    let ny = offset(y, dy, grid.height);
+
+    match (nx, ny) {
+        (Some(nx), Some(ny)) => Some((nx, ny)),
+        _ => None,
+    }
+}
+
+#[inline]
+fn offset(base: usize, offset: isize, max: usize) -> Option<usize> {
+    let res = base as isize + offset;
+
+    if res < 0 || res >= max as isize {
+        return None;
+    }
+
+    Some(res as usize)
+}

src/aoc2024/mod.rs

@@ -8,6 +8,7 @@ pub mod day2;
 pub mod day3;
 pub mod day4;
 pub mod day5;
+pub mod day6;
 
 pub fn run_solution_for_day(day: u32, input: &str, results: Option<Results>) -> Option<TimingData> {
     let r = results
@@ -20,6 +21,7 @@ pub fn run_solution_for_day(day: u32, input: &str, results: Option<Results>) ->
         3 => run::<Aoc2024, Day3>(input, r),
         4 => run::<Aoc2024, Day4>(input, r),
         5 => run::<Aoc2024, Day5>(input, r),
+        6 => run::<Aoc2024, Day6>(input, r),
         _ => return None,
     };
     Some(elapsed)