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

Refactored solutions putting common code in module

Browse Source 
Signed-off-by: Christopher Arndt <chris@chrisarndt.de>
This commit is contained in:
Christopher Arndt 2023-12-04 12:42:23 +01:00
parent aab15cdced
commit f725162eae
3 changed files with 44 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
day_04/common.nim Normal file
View File

@ -0,0 +1,32 @@
import std/[math, sequtils, setutils, strformat, strmisc, strutils]
# ASSUMPTION: lottery numbers are < 128
type
Card* = ref object
idx*: int
play, pot: set[int8]
proc value*(card: Card): int =
(card.play * card.pot).len
proc points*(card: Card): int =
result = case card.value:
of 0: 0
of 1: 1
else: 2 ^ (card.value - 1)
proc `$`*(card: Card): string =
return &"Card {card.idx+1}: {card.play}, {card.pot} => {card.value}"
proc parseNumberSet(s: string): set[int8] =
s.strip.splitWhitespace().mapIt(it.parseInt.int8).toSet
proc parseInput*(s: string): seq[Card] =
let lines = s.strip.splitLines
for i, line in lines:
let card = new Card
let (left, _, right) = line.split(":")[1].partition("|")
card.idx = i
card.play = parseNumberSet(left)
card.pot = parseNumberSet(right)
result.add(card)

36
day_04/solution_04a.nim
View File

@ -1,37 +1,11 @@
import std/[cmdline, math, sequtils, setutils, strformat, strscans, strutils]
import std/[cmdline, math, sequtils, strformat]
import common
var result = 0
var fn = if paramCount() > 0: paramStr(1) else: "input_04.txt"
var lines = readFile(fn).strip.splitLines
# 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
var cards = parseInput(readFile(fn))
var result = sum(cards.mapIt(it.points))
echo &"Cards processed: {cards.len}"
echo &"Result: {result}"

38
day_04/solution_04b.nim
View File

@ -1,36 +1,12 @@
import std/[cmdline, sequtils, setutils, strformat, strscans, strutils]
import std/[cmdline, strformat]
import common
var result = 0
var fn = if paramCount() > 0: paramStr(1) else: "input_04.txt"
var lines = readFile(fn).strip.splitLines
# ASSUMPTION: lottery numbers are < 128
type
Card = ref object
idx: int
play, pot: set[int8]
var cards: seq[Card]
proc value(card: Card): int =
(card.play * card.pot).len
#proc `$`(card: Card): string =
# return &"Card idx={card.idx}: value={card.value}"
for i, 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.idx = i
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 {card.idx}: {card.play}, {card.pot} => {card.value}"
cards.add(card)
var cards = parseInput(readFile(fn))
var stack = cards
var result = 0
while stack.len > 0:
result += len(stack)
var extra_cards: seq[Card]
@ -41,7 +17,7 @@ while stack.len > 0:
stack = extra_cards
#echo &"Cards processed: {cards.len}"
echo &"Cards processed: {cards.len}"
echo &"Result: {result}"
case fn: