chris/adventofcode-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor solution day 04 part 1

Browse Source 
Signed-off-by: Christopher Arndt <chris@chrisarndt.de>
This commit is contained in:
Christopher Arndt 2023-12-04 11:36:45 +01:00
parent 0f7658a8e0
commit 21b7041172
1 changed files with 28 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
day_04/solution_04a.nim
View File

@ -4,19 +4,35 @@ var result = 0
var fn = if paramCount() > 0: paramStr(1) else: "input_04.txt"
var lines = readFile(fn).strip.splitLines
for line in lines:
var card: int
let cardspec = line.split(":", 1)
if scanf(cardspec[0].strip, "Card$s$i", card):
let play = cardspec[1].split("|")[0].strip.splitWhitespace().mapIt(it.strip).mapIt(it.parseInt).mapIt(it.int8).toSet
let jackpot = cardspec[1].split("|")[1].strip.splitWhitespace().mapIt(it.strip).mapIt(it.parseInt).mapIt(it.int8).toSet
#echo &"Card {card}: {left}, {right} => {left * right}"
let wins = (play * jackpot).len
if wins == 1:
result += 1
elif wins > 1:
result += 2 ^ (wins - 1)
# ASSUMPTION: lottery numbers are < 128
type
Card = ref object
play, pot: set[int8]
var cards: seq[Card]
proc points(card: Card): int =
let matches = (card.play * card.pot).len
if matches == 1:
result = 1
elif matches > 1:
result = 2 ^ (matches - 1)
else:
result = 0
for line in lines:
var cardno: int
let cardspec = line.split(":", 1)
if scanf(cardspec[0].strip, "Card$s$i", cardno):
let card = new Card
card.play = cardspec[1].split("|")[0].strip.splitWhitespace().mapIt(it.strip).mapIt(it.parseInt).mapIt(it.int8).toSet
card.pot = cardspec[1].split("|")[1].strip.splitWhitespace().mapIt(it.strip).mapIt(it.parseInt).mapIt(it.int8).toSet
echo &"Card {cardno}: {card.play}, {card.pot} => {card.points}"
cards.add(card)
result += card.points
echo &"Cards processed: {cards.len}"
echo &"Result: {result}"
case fn: