AoC 2024 Day 10 - rust

Author: pareronia

README.md

@@ -10,7 +10,7 @@
 | ---| --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- |
 | python3 | [✓](src/main/python/AoC2024_01.py) | [✓](src/main/python/AoC2024_02.py) | [✓](src/main/python/AoC2024_03.py) | [✓](src/main/python/AoC2024_04.py) | [✓](src/main/python/AoC2024_05.py) | [✓](src/main/python/AoC2024_06.py) | [✓](src/main/python/AoC2024_07.py) | [✓](src/main/python/AoC2024_08.py) | [✓](src/main/python/AoC2024_09.py) | [✓](src/main/python/AoC2024_10.py) | [✓](src/main/python/AoC2024_11.py) | [✓](src/main/python/AoC2024_12.py) | [✓](src/main/python/AoC2024_13.py) | [✓](src/main/python/AoC2024_14.py) |  |  |  |  |  |  |  |  |  |  |  |
 | java | [✓](src/main/java/AoC2024_01.java) | [✓](src/main/java/AoC2024_02.java) | [✓](src/main/java/AoC2024_03.java) | [✓](src/main/java/AoC2024_04.java) | [✓](src/main/java/AoC2024_05.java) | [✓](src/main/java/AoC2024_06.java) | [✓](src/main/java/AoC2024_07.java) | [✓](src/main/java/AoC2024_08.java) |  | [✓](src/main/java/AoC2024_10.java) | [✓](src/main/java/AoC2024_11.java) | [✓](src/main/java/AoC2024_12.java) |  | [✓](src/main/java/AoC2024_14.java) |  |  |  |  |  |  |  |  |  |  |  |
-| rust | [✓](src/main/rust/AoC2024_01/src/main.rs) | [✓](src/main/rust/AoC2024_02/src/main.rs) | [✓](src/main/rust/AoC2024_03/src/main.rs) | [✓](src/main/rust/AoC2024_04/src/main.rs) | [✓](src/main/rust/AoC2024_05/src/main.rs) | [✓](src/main/rust/AoC2024_06/src/main.rs) | [✓](src/main/rust/AoC2024_07/src/main.rs) | [✓](src/main/rust/AoC2024_08/src/main.rs) |  |  |  |  | [✓](src/main/rust/AoC2024_13/src/main.rs) | [✓](src/main/rust/AoC2024_14/src/main.rs) |  |  |  |  |  |  |  |  |  |  |  |
+| rust | [✓](src/main/rust/AoC2024_01/src/main.rs) | [✓](src/main/rust/AoC2024_02/src/main.rs) | [✓](src/main/rust/AoC2024_03/src/main.rs) | [✓](src/main/rust/AoC2024_04/src/main.rs) | [✓](src/main/rust/AoC2024_05/src/main.rs) | [✓](src/main/rust/AoC2024_06/src/main.rs) | [✓](src/main/rust/AoC2024_07/src/main.rs) | [✓](src/main/rust/AoC2024_08/src/main.rs) |  | [✓](src/main/rust/AoC2024_10/src/main.rs) |  |  | [✓](src/main/rust/AoC2024_13/src/main.rs) | [✓](src/main/rust/AoC2024_14/src/main.rs) |  |  |  |  |  |  |  |  |  |  |  |
 <!-- @END:ImplementationsTable:2024@ -->
 
 ## 2023

src/main/rust/AoC2024_10/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "AoC2024_10"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+aoc = { path = "../aoc" }
+itertools = "0.11"

src/main/rust/AoC2024_10/src/main.rs

@@ -0,0 +1,97 @@
+#![allow(non_snake_case)]
+
+use aoc::grid::{Cell, Grid, IntGrid};
+use aoc::Puzzle;
+use itertools::Itertools;
+
+struct AoC2024_10;
+
+impl AoC2024_10 {
+    fn get_trails(&self, grid: &IntGrid) -> Vec<Vec<Cell>> {
+        fn dfs(grid: &IntGrid, trails: &mut Vec<Vec<Cell>>, trail: Vec<Cell>) {
+            if trail.len() == 10 {
+                trails.push(trail);
+                return;
+            }
+            let nxt = grid.get(trail.last().unwrap()) + 1;
+            for n in grid.capital_neighbours(trail.last().unwrap()) {
+                if grid.get(&n) == nxt {
+                    let mut new_trail = trail.to_vec();
+                    new_trail.push(n);
+                    dfs(grid, trails, new_trail);
+                }
+            }
+        }
+
+        let mut trails = vec![];
+        for s in grid.cells().filter(|cell| grid.get(cell) == 0) {
+            dfs(grid, &mut trails, vec![s]);
+        }
+        trails
+    }
+}
+
+impl aoc::Puzzle for AoC2024_10 {
+    type Input = IntGrid;
+    type Output1 = usize;
+    type Output2 = usize;
+
+    aoc::puzzle_year_day!(2024, 10);
+
+    fn parse_input(&self, lines: Vec<String>) -> Self::Input {
+        IntGrid::from(&lines.iter().map(AsRef::as_ref).collect::<Vec<_>>())
+    }
+
+    fn part_1(&self, grid: &Self::Input) -> Self::Output1 {
+        let trails = self.get_trails(grid);
+        trails
+            .iter()
+            .map(|trail| trail[0])
+            .unique()
+            .map(|zero| {
+                trails
+                    .iter()
+                    .filter(|trail| trail[0] == zero)
+                    .map(|trail| trail[9])
+                    .unique()
+                    .count()
+            })
+            .sum()
+    }
+
+    fn part_2(&self, grid: &Self::Input) -> Self::Output2 {
+        self.get_trails(grid).len()
+    }
+
+    fn samples(&self) {
+        aoc::puzzle_samples! {
+            self, part_1, TEST, 36,
+            self, part_2, TEST, 81
+        };
+    }
+}
+
+fn main() {
+    AoC2024_10 {}.run(std::env::args());
+}
+
+const TEST: &str = "\
+89010123
+78121874
+87430965
+96549874
+45678903
+32019012
+01329801
+10456732
+";
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    pub fn samples() {
+        AoC2024_10 {}.samples();
+    }
+}