adventofcode/2021/8/first.nim

24 lines
557 B
Nim

import
../../adventofcode,
math,
sequtils,
strformat,
strutils,
tables
let lines = adventofcode.getInput(2021, 8).splitLines()
var decodedDigits = newTable[int, int]()
for i in [2, 3, 4, 7]:
decodedDigits[i] = 0
for line in lines:
let lineParts = line.split(" | ")
# We only consider output digits for now
let outputDigits = lineParts[1].split(" ")
for outputDigit in outputDigits:
if decodedDigits.hasKey outputDigit.len():
decodedDigits[outputDigit.len()] += 1
echo fmt"Answer: {math.sum toSeq decodedDigits.values()}"