From 548fea505d0e7b461ece094e574340582e4c19eb Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Sun, 3 Dec 2023 10:56:47 +0100
Subject: [PATCH] day 3 puzzle 1

---
 src/day3/Puzzle1.scala       | 87 ++++++++++++++++++++++++++++++++++++
 tests/day3/Puzzle1Test.scala |  9 ++++
 tests_res/day3/input1.txt    | 10 +++++
 3 files changed, 106 insertions(+)
 create mode 100644 src/day3/Puzzle1.scala
 create mode 100644 tests/day3/Puzzle1Test.scala
 create mode 100644 tests_res/day3/input1.txt

diff --git a/src/day3/Puzzle1.scala b/src/day3/Puzzle1.scala
new file mode 100644
index 0000000..04f0fcc
--- /dev/null
+++ b/src/day3/Puzzle1.scala
@@ -0,0 +1,87 @@
+package day3
+
+import scala.io.{BufferedSource, Source}
+
+object Puzzle1 {
+  val OFFSETS: Array[Array[Int]] = Array(
+    Array(-1,-1), Array( 0,-1), Array( 1,-1),
+    Array(-1, 0),               Array( 1, 0),
+    Array(-1, 1), Array( 0, 1), Array( 1, 1)
+  )
+  var width = 0
+  var height = 0
+
+  var schematic: Array[Array[Char]] = Array.empty
+  def loadInput(path: String): Unit = {
+    val source: BufferedSource = Source.fromFile(path)
+    val lines: Array[String] = source.getLines().toArray
+    height = lines.length
+    width = if (height == 0) 0 else lines(0).length
+    schematic = new Array(height)
+
+    var i = 0
+    for (line: String <- lines) {
+      schematic(i) = line.split("").map(s => s(0))
+      i += 1
+    }
+
+    source.close()
+  }
+
+  def getNumber(sx: Int, sy: Int): String = {
+    var x: Int = sx
+    var number: String = ""
+    var c: Char = schematic(sy)(sx)
+    do {
+      number += c
+      x += 1
+      if (x < width) c = schematic(sy)(x)
+    } while ('0' <= c && c <= '9' && x < width)
+    return number
+  }
+
+  def isPartNumber(sx: Int, sy: Int, len: Int): Boolean = {
+    for (x: Int <- sx until sx+len) {
+      for (offset: Array[Int] <- OFFSETS) {
+        val x2: Int =  x + offset(0)
+        val y2: Int = sy + offset(1)
+        if (0 <= x2 && x2 < width && 0 <= y2 && y2 < height) {
+          val c: Char = schematic(y2)(x2)
+          if (c != '.' && (c < '0' || c > '9')) {
+            return true
+          }
+        }
+      }
+    }
+    return false
+  }
+
+  def solve(path: String): Int = {
+    loadInput(path)
+
+    var sum: Int = 0
+    var y: Int = 0
+    var x: Int = 0
+
+    while (y < schematic.length) {
+      while (x < schematic(y).length) {
+        val c: Char = schematic(y)(x)
+        if ('0' <= c && c <= '9') {
+          val number: String = getNumber(x, y)
+          if (isPartNumber(x, y, number.length)) {
+            sum += number.toInt
+            x += number.length - 1
+          }
+        }
+        x += 1
+      }
+      y += 1
+      x = 0
+    }
+    return sum
+  }
+  def main(args: Array[String]): Unit = {
+    val solution: Int = solve("res/day3/input1.txt")
+    println(solution)
+  }
+}
diff --git a/tests/day3/Puzzle1Test.scala b/tests/day3/Puzzle1Test.scala
new file mode 100644
index 0000000..02d232c
--- /dev/null
+++ b/tests/day3/Puzzle1Test.scala
@@ -0,0 +1,9 @@
+package day3
+
+import org.scalatest.funsuite.AnyFunSuite
+
+class Puzzle1Test extends AnyFunSuite {
+  test("Puzzle1.solve") {
+    assert(Puzzle1.solve("tests_res/day3/input1.txt") == 4361)
+  }
+}
diff --git a/tests_res/day3/input1.txt b/tests_res/day3/input1.txt
new file mode 100644
index 0000000..624ea4f
--- /dev/null
+++ b/tests_res/day3/input1.txt
@@ -0,0 +1,10 @@
+467..114..
+...*......
+..35..633.
+......#...
+617*......
+.....+.58.
+..592.....
+......755.
+...$.*....
+.664.598..
\ No newline at end of file