mirror of
https://github.com/daylinmorgan/advent-of-code-2023.git
synced 2024-11-13 15:17:53 -06:00
70 lines
1.8 KiB
Nim
70 lines
1.8 KiB
Nim
import std/[strutils, sugar]
|
|
|
|
const example* = slurp("example.txt").strip()
|
|
const input* = slurp("input.txt").strip()
|
|
|
|
type
|
|
Turn = object
|
|
cubes: seq[(int, string)]
|
|
|
|
Game = object
|
|
turns: seq[Turn]
|
|
|
|
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()
|
|
for cube in turn.split(","):
|
|
let s = cube.strip().split()
|
|
result[^1].turns[^1].cubes.add (parseInt(s[0]),s[1])
|
|
|
|
proc isValid(turn: Turn): bool =
|
|
const limits = (red: 12, green: 13, blue: 14)
|
|
for cube in turn.cubes:
|
|
case cube[1]:
|
|
of "red":
|
|
if cube[0] > limits.red: 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(
|
|
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)
|
|
for turn in game.turns:
|
|
for cube in turn.cubes:
|
|
case cube[1]:
|
|
of "red": mins.red = max(mins.red, cube[0])
|
|
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
|
|
check partOne(input) == 2239
|
|
test "part two":
|
|
check partTwo(example) == 2286
|
|
check partTwo(input) == 83435
|
|
|