From ad788f475802851a79259aa080b9864c2b2e5e7e Mon Sep 17 00:00:00 2001
From: Natsu Kagami <nki@nkagami.me>
Date: Sat, 2 Dec 2023 15:31:39 +0100
Subject: [PATCH] Day 2

---
 Common.scala                    |   6 ++
 Day1.scala                      |  17 ++----
 Day2.scala                      |  51 ++++++++++++++++
 day1.input => inputs/day1.input |   0
 inputs/day2.input               | 100 ++++++++++++++++++++++++++++++++
 run                             |  21 +++++++
 6 files changed, 182 insertions(+), 13 deletions(-)
 create mode 100644 Common.scala
 create mode 100644 Day2.scala
 rename day1.input => inputs/day1.input (100%)
 create mode 100644 inputs/day2.input
 create mode 100755 run

diff --git a/Common.scala b/Common.scala
new file mode 100644
index 0000000..982e275
--- /dev/null
+++ b/Common.scala
@@ -0,0 +1,6 @@
+package aoc
+
+def iterate[T](get: => T | Null) =
+  Iterator.continually(get).takeWhile(_ != null).map(_.nn)
+
+def lines = iterate(scala.io.StdIn.readLine())
diff --git a/Day1.scala b/Day1.scala
index c9106e0..64d53fb 100644
--- a/Day1.scala
+++ b/Day1.scala
@@ -1,5 +1,7 @@
 // Part 1
 
+import aoc._
+
 def literalDigits(s: String) =
   s.toSeq.filter(_.isDigit).map(v => (v - '0').toInt)
 
@@ -7,20 +9,9 @@ def getValue(toDigits: String => Seq[Int])(s: String): Int =
   val digits = toDigits(s)
   digits.head * 10 + digits.last
 
-def iterate[T](get: => T | Null) = new scala.collection.IterableOnce[T]:
-  override def iterator = new Iterator[T]:
-    var item: Option[T | Null] = None
-    override def hasNext: Boolean =
-      if item.isEmpty then item = Some(get)
-      item.get != null
-    override def next(): T =
-      val r = item.get.nn
-      item = None
-      r
-
 def getSum(toDigits: String => Seq[Int]) =
   val values =
-    Seq.from(iterate(scala.io.StdIn.readLine())).map(getValue(toDigits))
+    lines.map(getValue(toDigits))
   println(values.sum)
 
 // Part 2
@@ -40,6 +31,6 @@ def literalOrEnglishDigits(s: String) =
       loop(idx - 1, nAccum)
   loop(s.length() - 1, Seq.empty)
 
-@main def Run(part: Int) = part match
+@main def Day1(part: Int) = part match
   case 1 => getSum(literalDigits)
   case 2 => getSum(literalOrEnglishDigits)
diff --git a/Day2.scala b/Day2.scala
new file mode 100644
index 0000000..c886123
--- /dev/null
+++ b/Day2.scala
@@ -0,0 +1,51 @@
+//> using dep org.scala-lang.modules::scala-parser-combinators::2.3.0
+
+import aoc._
+import scala.util.parsing.combinator._
+
+case class Hand(red: Int, green: Int, blue: Int):
+  def max(other: Hand) =
+    Hand(red max other.red, green max other.green, blue max other.blue)
+  def power = red * green * blue
+case class Game(id: Int, hands: List[Hand]):
+  def power =
+    hands.foldLeft(Hand(0, 0, 0))(_ max _).power
+
+object Parser extends RegexParsers:
+  val num = """(0|[1-9]\d*)""".r ^^ { _.toInt }
+  val update =
+    num ~ """(red|green|blue)""".r ^^ { case (num ~ color) =>
+      color match
+        case "red"   => (h: Hand) => h.copy(red = num)
+        case "green" => (h: Hand) => h.copy(green = num)
+        case "blue"  => (h: Hand) => h.copy(blue = num)
+    }
+  val hand = rep1sep(update, ",").map(_.foldRight(Hand(0, 0, 0))(_(_)))
+
+  val game = "Game" ~ num ~ ":" ~ repsep(hand, ";") ^^ {
+    case (_ ~ num ~ _ ~ hands) => Game(num, hands)
+  }
+end Parser
+
+// Part 1
+
+def getGames = lines.map(Parser.parse(Parser.game, _).get)
+
+def part1 =
+  val games = getGames.toList
+  val res =
+    games
+      .filter(g =>
+        g.hands.forall(h => h.red <= 12 && h.green <= 13 && h.blue <= 14)
+      )
+      .map(_.id)
+      .sum
+  println(res)
+
+def part2 =
+  val res = getGames.map(_.power).sum
+  println(res)
+
+@main def Day2(part: Int) = part match
+  case 1 => part1
+  case 2 => part2
diff --git a/day1.input b/inputs/day1.input
similarity index 100%
rename from day1.input
rename to inputs/day1.input
diff --git a/inputs/day2.input b/inputs/day2.input
new file mode 100644
index 0000000..f6238c6
--- /dev/null
+++ b/inputs/day2.input
@@ -0,0 +1,100 @@
+Game 1: 4 blue; 1 green, 2 red; 4 blue, 1 green, 6 red
+Game 2: 7 red, 1 green, 4 blue; 13 red, 11 blue; 6 red, 2 blue; 9 blue, 9 red; 4 blue, 11 red; 15 red, 1 green, 3 blue
+Game 3: 1 blue, 10 green; 4 green, 8 blue; 3 blue, 14 green, 1 red
+Game 4: 1 green, 2 blue; 1 blue, 4 green; 8 red, 3 blue, 3 green; 8 red, 2 green, 1 blue; 7 green, 3 blue, 2 red; 1 red, 1 green, 3 blue
+Game 5: 3 red, 7 blue, 4 green; 12 blue, 16 red, 4 green; 9 red, 2 green; 1 blue, 1 green, 1 red
+Game 6: 15 blue; 15 red, 14 blue, 2 green; 8 red, 2 blue, 2 green; 2 green, 11 blue, 1 red
+Game 7: 6 green, 6 red, 2 blue; 1 blue, 2 red, 7 green; 12 red; 5 green, 3 red, 1 blue; 16 red, 10 green
+Game 8: 2 green, 10 red, 3 blue; 1 blue, 5 green, 11 red; 6 red, 1 blue, 2 green; 11 red; 4 red, 1 blue, 5 green; 5 green, 3 blue
+Game 9: 5 blue, 11 red; 2 blue, 2 green; 11 red, 2 green
+Game 10: 3 red, 5 green; 3 blue, 5 green; 3 red, 2 blue, 14 green
+Game 11: 1 green, 1 blue, 6 red; 2 blue, 7 red, 2 green; 2 green, 2 red, 3 blue; 10 red; 2 red, 2 blue; 11 red, 3 blue
+Game 12: 6 blue, 8 red, 6 green; 15 green, 4 red; 1 red, 10 green, 1 blue; 1 red, 3 blue, 11 green
+Game 13: 2 blue, 6 red; 15 red, 6 blue; 20 blue, 10 red, 3 green; 17 blue, 1 red, 4 green
+Game 14: 3 green, 7 blue, 7 red; 2 blue; 7 blue, 10 red
+Game 15: 3 green, 7 blue; 9 green, 8 blue, 5 red; 6 green, 13 red; 14 red, 2 green, 15 blue; 15 red, 7 green
+Game 16: 13 blue; 6 green, 4 blue, 11 red; 15 red, 2 green, 6 blue; 1 green, 13 red, 8 blue; 8 green, 7 blue, 14 red
+Game 17: 3 red; 17 red, 4 green; 1 blue, 11 red; 3 blue, 20 red, 3 green
+Game 18: 4 red, 2 blue, 3 green; 9 red, 6 green; 11 red, 1 blue
+Game 19: 1 green, 4 blue; 1 green, 2 red; 2 blue, 1 green; 4 red, 2 blue
+Game 20: 15 blue, 6 green, 6 red; 13 green, 1 blue, 1 red; 13 green, 13 blue, 5 red; 7 red, 16 green, 4 blue
+Game 21: 10 blue, 5 green, 8 red; 6 blue, 12 red, 4 green; 2 green, 16 blue, 3 red; 6 red, 12 blue, 3 green; 1 red, 3 green; 7 blue, 6 green, 7 red
+Game 22: 3 blue, 3 green, 17 red; 1 green, 18 red; 3 green, 10 red, 5 blue; 2 green, 2 red, 4 blue; 2 blue, 13 red; 1 green, 3 blue, 9 red
+Game 23: 4 green, 16 red, 2 blue; 10 red, 10 green, 7 blue; 2 green, 6 red, 13 blue; 1 green, 13 blue, 16 red; 7 green, 16 blue, 9 red
+Game 24: 6 blue, 7 red, 4 green; 6 blue, 2 green; 2 green, 6 blue, 2 red; 5 red, 3 green, 11 blue
+Game 25: 4 red, 2 green; 1 green; 4 green, 4 blue, 8 red; 5 red, 1 blue
+Game 26: 9 red, 4 blue, 13 green; 3 blue, 10 red, 7 green; 13 blue, 5 green, 9 red
+Game 27: 1 green, 12 red, 2 blue; 2 blue, 13 red, 2 green; 2 blue, 7 red; 4 green, 9 red, 2 blue; 1 blue, 2 green, 15 red; 3 red, 4 green, 1 blue
+Game 28: 1 red, 9 blue, 17 green; 14 green, 15 blue, 2 red; 4 red, 18 green, 13 blue
+Game 29: 16 green, 5 blue, 1 red; 6 green, 6 red, 16 blue; 4 red, 9 green, 12 blue; 5 green, 14 blue, 1 red
+Game 30: 3 red, 2 blue, 12 green; 13 green, 4 red; 13 green, 2 red, 1 blue; 2 blue, 6 red, 4 green; 3 blue, 13 green, 5 red
+Game 31: 3 red; 6 red, 2 green; 5 red; 3 green, 2 red; 1 green, 2 red, 1 blue; 1 blue, 6 red
+Game 32: 1 red, 7 green; 9 green, 5 blue; 1 green, 2 red; 4 blue, 2 red, 1 green; 4 blue, 1 green, 3 red
+Game 33: 11 green; 12 blue, 2 green; 5 green, 1 blue; 10 green, 3 blue; 4 blue, 1 red, 4 green; 4 green, 5 blue
+Game 34: 4 red, 8 blue, 2 green; 8 green, 4 red, 14 blue; 11 green, 6 red, 8 blue; 16 green, 3 blue, 5 red; 3 blue, 3 red, 13 green
+Game 35: 7 green, 12 red, 1 blue; 1 red; 13 red; 14 red, 2 blue, 9 green
+Game 36: 3 red, 4 green, 1 blue; 3 red, 4 blue; 6 red, 4 blue, 3 green; 3 green, 4 blue, 3 red; 2 blue, 4 green, 7 red
+Game 37: 2 green, 1 blue, 5 red; 1 green; 3 blue; 3 blue, 1 green
+Game 38: 1 red, 12 blue, 17 green; 4 blue, 2 red, 8 green; 7 blue, 20 green; 6 red, 3 blue; 6 green, 7 red, 6 blue; 10 green, 3 red
+Game 39: 3 green, 3 blue, 2 red; 4 blue, 4 red, 4 green; 4 blue, 4 red; 1 blue, 5 green, 2 red; 5 green, 3 blue, 4 red; 4 green, 2 blue
+Game 40: 18 green, 1 red; 17 green, 1 blue; 2 green, 1 blue, 1 red; 9 green, 1 blue; 3 green, 1 red; 1 red, 10 green
+Game 41: 2 red, 4 blue, 3 green; 8 blue, 2 red; 5 blue; 2 green, 2 red, 3 blue; 1 green, 7 blue
+Game 42: 1 green, 2 blue; 9 green, 2 blue, 15 red; 1 green, 4 blue, 9 red
+Game 43: 5 blue, 3 red; 2 blue, 8 red, 7 green; 17 red, 4 blue, 7 green
+Game 44: 13 red, 3 green, 12 blue; 15 green, 10 blue; 8 green, 11 red, 2 blue; 10 blue, 16 red, 2 green; 12 blue, 5 green, 5 red; 14 green, 8 red, 13 blue
+Game 45: 1 red, 3 green; 4 green, 5 blue, 2 red; 6 red, 2 blue, 6 green; 3 blue, 2 green; 5 blue, 3 green, 4 red; 5 red, 5 blue, 6 green
+Game 46: 12 red, 2 blue, 3 green; 15 red, 14 blue, 11 green; 6 red, 11 blue, 6 green; 4 red, 1 green; 7 blue, 14 red; 14 red, 18 blue, 6 green
+Game 47: 3 blue, 5 red, 4 green; 1 blue, 10 red; 6 blue, 5 green, 7 red; 3 red, 4 green; 2 blue, 2 green, 13 red; 4 blue, 13 red, 2 green
+Game 48: 2 green, 3 blue, 7 red; 12 red, 1 green, 2 blue; 5 red, 2 blue; 4 blue, 3 green, 10 red
+Game 49: 8 green, 13 blue, 3 red; 14 blue, 1 green; 14 blue, 2 green
+Game 50: 1 red, 2 green, 3 blue; 2 green, 2 red; 1 green, 5 blue; 4 green
+Game 51: 10 green, 5 red; 10 green, 2 blue, 2 red; 2 blue, 13 red, 1 green; 6 blue, 10 green, 3 red
+Game 52: 8 green, 1 blue, 6 red; 4 green, 5 blue; 4 green, 7 red; 3 blue, 6 green, 3 red; 7 red, 6 blue, 7 green; 4 red, 8 green, 4 blue
+Game 53: 11 blue, 10 green, 1 red; 6 blue, 1 green, 12 red; 6 green, 12 blue, 13 red; 1 blue, 10 green, 10 red; 11 green, 2 blue; 7 green, 7 red, 5 blue
+Game 54: 3 blue, 1 green, 7 red; 18 blue, 3 red, 1 green; 11 blue, 6 red
+Game 55: 9 blue, 1 red; 3 blue, 1 green, 2 red; 1 green, 6 blue, 5 red; 1 green, 5 red, 12 blue; 5 red, 3 green, 12 blue; 12 blue
+Game 56: 3 red, 1 green, 11 blue; 2 red, 20 blue; 12 blue, 4 red; 3 red, 2 blue, 6 green
+Game 57: 1 green, 13 red, 1 blue; 7 green, 2 red, 2 blue; 6 red, 3 blue; 6 blue, 4 red, 3 green; 1 green, 11 red
+Game 58: 3 red, 13 blue, 2 green; 6 green, 6 red, 19 blue; 4 blue, 9 green, 1 red; 1 blue, 6 red
+Game 59: 11 red, 2 blue, 2 green; 1 blue, 13 red; 12 red, 6 blue
+Game 60: 8 blue, 4 red, 11 green; 10 green; 5 blue, 3 red, 8 green; 6 blue, 6 red, 12 green
+Game 61: 1 green, 1 blue, 3 red; 1 blue, 2 green, 5 red; 4 red, 1 green, 1 blue; 5 red, 2 green
+Game 62: 14 blue, 2 green, 11 red; 11 red, 2 green, 8 blue; 5 blue, 14 red, 5 green; 17 red, 2 blue, 3 green; 2 red, 3 green, 5 blue; 11 blue, 10 red, 3 green
+Game 63: 2 blue, 2 green; 9 blue, 3 red; 1 green, 2 red, 12 blue
+Game 64: 14 green, 1 blue, 5 red; 4 red, 14 green, 12 blue; 10 blue, 3 red, 10 green
+Game 65: 1 green, 6 red, 6 blue; 7 red, 7 blue, 3 green; 14 blue, 5 red
+Game 66: 10 blue, 2 red, 7 green; 3 red, 16 blue; 10 green, 7 red, 17 blue; 10 red, 5 green, 5 blue; 13 blue, 10 green, 6 red
+Game 67: 9 blue, 6 green; 1 red, 8 blue, 9 green; 3 blue, 1 green, 1 red; 2 blue, 6 green, 1 red
+Game 68: 4 green, 9 red, 3 blue; 6 blue, 5 green, 2 red; 6 blue, 9 red, 3 green; 4 red, 2 green; 4 red, 9 green
+Game 69: 1 green, 1 blue, 2 red; 2 red, 7 green; 3 red, 1 blue, 5 green; 8 red, 7 green; 2 green, 1 blue; 6 red, 1 blue, 7 green
+Game 70: 13 blue, 3 green, 5 red; 1 red, 1 green, 6 blue; 4 red, 11 blue; 14 blue, 5 red, 1 green; 8 red, 16 blue, 1 green
+Game 71: 1 blue, 1 green; 6 blue, 2 red; 5 green, 1 red, 4 blue; 4 green, 3 red
+Game 72: 4 green, 2 blue, 11 red; 4 red, 7 green, 4 blue; 3 red, 6 green, 14 blue; 4 green, 12 red, 15 blue; 4 blue, 14 red; 6 blue, 13 red, 6 green
+Game 73: 4 green, 6 red, 7 blue; 11 red, 4 blue, 6 green; 8 red, 2 blue, 5 green; 3 red, 1 green, 7 blue
+Game 74: 5 blue, 10 green; 6 green, 5 blue, 10 red; 4 green, 2 red, 1 blue; 3 blue, 11 green
+Game 75: 3 red, 3 green, 15 blue; 6 blue, 3 green, 5 red; 11 blue, 1 red, 3 green; 7 blue, 3 green, 4 red; 9 blue, 1 red, 3 green
+Game 76: 11 red; 7 green, 12 red; 2 red, 1 blue, 2 green; 2 red, 1 blue, 6 green; 5 red, 7 green; 1 blue, 8 green
+Game 77: 2 blue, 15 green, 1 red; 6 blue, 1 red; 1 green, 5 blue, 1 red; 2 blue, 1 red, 1 green; 15 green, 8 blue, 1 red; 19 green, 5 blue
+Game 78: 14 red, 2 green, 7 blue; 2 green, 14 red, 3 blue; 1 blue, 7 red
+Game 79: 15 red, 2 green, 1 blue; 3 red, 1 green; 12 red, 2 blue; 12 red, 1 green; 1 blue, 2 red, 1 green
+Game 80: 2 red, 1 green, 7 blue; 7 red, 6 blue, 5 green; 6 blue, 6 red; 6 green, 2 blue, 3 red; 5 red, 5 blue, 1 green
+Game 81: 10 red, 1 green, 3 blue; 6 green, 13 blue, 3 red; 1 green, 2 red, 10 blue
+Game 82: 4 blue, 1 red, 7 green; 4 red, 14 blue, 8 green; 1 red, 11 blue, 6 green
+Game 83: 10 red, 3 blue, 9 green; 3 green, 3 red, 1 blue; 4 blue, 11 green, 8 red; 2 blue, 8 green, 2 red; 2 green, 2 red
+Game 84: 2 green, 2 blue, 14 red; 7 red, 5 blue, 11 green; 4 red, 6 blue, 5 green; 3 blue, 13 green, 14 red; 6 red, 7 blue, 8 green; 2 blue, 3 red, 18 green
+Game 85: 8 green, 14 blue; 6 green, 9 red, 15 blue; 9 red, 12 green, 15 blue; 12 green, 6 red; 9 green, 10 red, 15 blue; 12 blue, 6 green
+Game 86: 1 blue, 1 green, 4 red; 6 green, 4 red, 6 blue; 1 red, 4 blue, 4 green; 6 green, 2 blue, 1 red
+Game 87: 17 blue, 13 green; 8 blue, 3 red; 16 green, 4 red, 6 blue
+Game 88: 11 red, 16 blue, 6 green; 10 red, 2 blue, 1 green; 5 blue, 2 green, 14 red
+Game 89: 3 blue, 2 green; 2 red; 9 blue, 8 green, 1 red; 2 green, 2 blue, 3 red; 4 red, 3 green
+Game 90: 2 blue, 14 red, 2 green; 6 blue, 2 red, 2 green; 17 red, 1 blue, 6 green; 1 blue, 8 green, 1 red
+Game 91: 6 green, 1 blue, 13 red; 10 red, 4 green, 12 blue; 9 green, 17 red, 3 blue; 12 blue, 5 red, 2 green; 2 green, 9 red, 14 blue
+Game 92: 2 red, 4 green, 6 blue; 9 blue, 3 green, 6 red; 5 blue, 4 green; 3 blue, 2 green, 7 red; 4 red, 4 green, 11 blue
+Game 93: 4 red, 11 blue, 9 green; 10 blue, 3 green, 9 red; 3 green, 11 red, 1 blue
+Game 94: 11 green, 3 red, 1 blue; 3 green, 2 red, 6 blue; 2 red, 6 blue, 5 green; 4 blue, 5 green, 5 red; 17 green, 6 red, 6 blue; 5 green, 6 red, 7 blue
+Game 95: 1 red, 3 blue, 15 green; 5 green, 6 blue; 11 green, 2 red, 11 blue; 15 green, 5 red, 7 blue
+Game 96: 13 red, 3 blue; 3 red, 13 blue; 5 blue, 1 red, 2 green; 7 red, 7 blue; 12 red, 9 blue, 3 green; 8 red, 15 blue, 2 green
+Game 97: 4 blue, 9 green, 2 red; 2 red, 5 green, 13 blue; 9 blue, 2 red, 16 green
+Game 98: 3 red; 1 green, 10 red; 2 blue, 8 red; 1 green, 11 red, 2 blue
+Game 99: 6 red, 14 green; 8 green, 15 red; 1 red, 4 green; 2 blue, 7 green, 13 red; 14 green, 5 red, 1 blue; 1 blue, 5 red, 8 green
+Game 100: 9 blue, 18 green, 4 red; 5 green, 10 blue, 11 red; 1 green, 1 red; 16 green, 5 red, 1 blue
diff --git a/run b/run
new file mode 100755
index 0000000..76092ed
--- /dev/null
+++ b/run
@@ -0,0 +1,21 @@
+#!/usr/bin/env fish
+
+if test (count $argv) -lt 2
+    echo "$(status -f) [day] [part] ..."
+    exit 1
+end
+
+set input ""
+if test $argv[1] = "--with-input"
+    set input "yes"
+    set -e argv[1]
+end
+
+set day $argv[1]
+set part $argv[2]
+
+if test -n $input
+    scala-cli run **.scala --main-class "Day$day" -- $part $argv[3..] < ./inputs/day$day.input
+else
+    scala-cli run **.scala --main-class "Day$day" -- $part $argv[3..]
+end