Day 04 part two
Signed-off-by: Christopher Arndt <chris@chrisarndt.de>
This commit is contained in:
		
							parent
							
								
									21b7041172
								
							
						
					
					
						commit
						aab15cdced
					
				
							
								
								
									
										51
									
								
								day_04/solution_04b.nim
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								day_04/solution_04b.nim
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,51 @@
 | 
			
		||||
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
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user