52 lines
1.4 KiB
Nim
52 lines
1.4 KiB
Nim
import std/[cmdline, sequtils, setutils, strformat, strscans, strutils]
|
|
|
|
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 stack = cards
|
|
while stack.len > 0:
|
|
result += len(stack)
|
|
var extra_cards: seq[Card]
|
|
|
|
for card in stack:
|
|
if card.value > 0:
|
|
extra_cards.add(cards[card.idx + 1..min(card.idx + card.value, cards.len - 1)])
|
|
|
|
stack = extra_cards
|
|
|
|
#echo &"Cards processed: {cards.len}"
|
|
echo &"Result: {result}"
|
|
|
|
case fn:
|
|
of "input_04.txt":
|
|
assert result == 11787590
|
|
of "sample_input_04.txt":
|
|
assert result == 30
|