use scanf+tuples+arrays for day 2

solutions/day02/solution.nim

@@ -1,65 +1,55 @@
-import std/[strutils, sugar]
+import std/[strutils, strscans]
 
 const example* = slurp("example.txt").strip()
 const input* = slurp("input.txt").strip()
 
 type
-  Turn = object
+  Color {.pure.} = enum
-    cubes: seq[(int, string)]
+    red, blue, green
 
+  Cubes = array[Color, int]
   Game = object
-    turns: seq[Turn]
+    turns: seq[Cubes]
 
 proc parseInput(input: string): seq[Game] =
   for line in input.splitLines:
     result.add Game()
     let turns = line.split(":")[1]
     for turn in turns.split(";"):
-      result[^1].turns.add Turn()
+      var cubes: Cubes
-      for cube in turn.split(","):
+      for cubeStr in turn.split(","):
-        let s = cube.strip().split()
+        var cnt: int
-        result[^1].turns[^1].cubes.add (parseInt(s[0]),s[1])
+        var color: string
+        discard cubeStr.strip().scanf("$i $w", cnt, color)
+        cubes[parseEnum[Color](color)] = cnt
+      result[^1].turns.add cubes
 
-proc isValid(turn: Turn): bool =
+proc isValid(game: Game): bool =
-  const limits = (red: 12, green: 13, blue: 14)
+  const limits: Cubes = [red: 12, blue: 14, green: 13]
-  for cube in turn.cubes:
+  for turn in game.turns:
-    case cube[1]:
+    for col, cnt in turn:
-      of "red":
+      if cnt > limits[col]:
-        if cube[0] > limits.red: return false
+        return false
-      of "green":
-        if cube[0] > limits.green: return false
-      of "blue":
-        if cube[0] > limits.blue: return false
-      else: discard
   return true
 
 proc partOne*(input: string): int =
   var games = parseInput(input)
   for i, game in games:
-    if collect(
+    if game.isValid():
-      for turn in game.turns:
-        if not turn.isValid: turn).len == 0:
       result += (i+1)
 
-
 proc partTwo*(input: string): int =
   var games = parseInput(input)
   for i, game in games:
-    var mins = (red: 0, green: 0, blue: 0)
+    var mins: Cubes
     for turn in game.turns:
-      for cube in turn.cubes:
+      for col, cnt in turn:
-        case cube[1]:
+        mins[col] = max(mins[col], cnt)
-          of "red": mins.red = max(mins.red, cube[0])
+    result += mins[red] * mins[green] * mins[blue]
-          of "green": mins.green = max(mins.green, cube[0])
-          of "blue": mins.blue = max(mins.blue, cube[0])
-          else: discard
-    result += mins.red * mins.green * mins.blue
-
 
 
 when isMainModule:
   import std/unittest
-
   suite "day 2":
     test "part one":
       check partOne(example) == 8
@@ -67,4 +57,3 @@ when isMainModule:
     test "part two":
       check partTwo(example) == 2286
       check partTwo(input) == 83435
-