finish day 3 part 1

src/bin/day3.rs

@@ -1,5 +1,5 @@
 use aoc23::prelude::*;
-use std::fmt::Display;
+use std::str::FromStr;
 
 #[derive(Debug)]
 enum Item {
@@ -18,6 +18,7 @@ struct Number {
 
 struct Grid {
     pub rows: Vec<Vec<Item>>,
+    pub numbers: Vec<Number>,
 }
 
 impl Grid {
@@ -48,25 +49,17 @@ impl Grid {
         }
         return items;
     }
-
-    fn print(&self) {
-        for row in self.rows.iter() {
-            for col in row {
-                match col {
-                    Item::Digit(_) => print!("d"),
-                    Item::Symbol => print!("s"),
-                    Item::None => print!("."),
-                }
-            }
-            println!();
-        }
-    }
 }
 
-fn part1(input: String) -> Result<impl Display> {
-    let mut grid = Grid { rows: vec![] };
-    let mut numbers: Vec<Number> = vec![];
-    for (row_index, line) in input.lines().enumerate() {
+impl FromStr for Grid {
+    type Err = BoxE;
+
+    fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
+        let mut grid = Grid {
+            rows: vec![],
+            numbers: vec![],
+        };
+        for (row_index, line) in s.lines().enumerate() {
             let mut row: Vec<Item> = vec![];
             let mut num: Option<usize> = None;
             let mut char_start = 0;
@@ -88,7 +81,7 @@ fn part1(input: String) -> Result<impl Display> {
                         row.push(Item::Symbol)
                     }
                     if let Some(value) = num {
-                    numbers.push(Number {
+                        grid.numbers.push(Number {
                             value,
                             row_index,
                             char_start,
@@ -99,23 +92,40 @@ fn part1(input: String) -> Result<impl Display> {
                     }
                 }
             }
+            if let Some(value) = num {
+                grid.numbers.push(Number {
+                    value,
+                    row_index,
+                    char_start,
+                    char_end: line.len() - 1,
+                });
+            }
             grid.rows.push(row);
         }
-    let grid = grid;
+        Ok(grid)
+    }
+}
+
+fn part1(input: String) -> Result<usize> {
+    let grid: Grid = input.parse()?;
     let mut sum = 0;
-    for number in numbers.iter() {
+    for number in grid.numbers.iter() {
         let adjacent =
             grid.get_adjacent_items(number.row_index, number.char_start, number.char_end);
-        let is_part_no = adjacent.iter().any(|item| !matches!(item, Item::None));
+        let is_part_no = adjacent.iter().any(|item| matches!(item, Item::Symbol));
+
+        // ```
         // println!(
         //     "Checking number: {:?}, is part: {}\tadjacent: {:?}",
         //     number.value, is_part_no, adjacent
         // );
+        // ```
         if is_part_no {
-            print!("{}, ", number.value);
+            // print!("{}, ", number.value);
             sum += number.value;
         }
     }
+
     // grid.print();
     Ok(sum)
 }
@@ -125,14 +135,19 @@ async fn main() -> Result<()> {
     println!("Day 3");
     println!("=====");
     let input = utils::aoc::get_puzzle_input(3).await?;
-
-    // ```
-    // let input = "467..114..
-    // ```
-    //
-    // ..._...... ..35..633. ......#... 617_...... .....+.58. ..592..... ......755.
-    // ...$.*.... .664.598.." .to_owned();
-    println!("part 1: {}", part1(input.clone())?);
+    let input = "467..114..
+...*......
+..35..633.
+......#...
+617*......
+.....+.58.
+..592.....
+......755.
+...$.*....
+.664.598.."
+        .to_owned();
+    let part_1 = part1(input.clone())?;
+    println!("part 1: {} == 520019 !! {}", part_1, part_1 == 520019);
 
     // println!("part 2: {}", part2(input.clone())?);
     Ok(())