You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
575 B
Racket
20 lines
575 B
Racket
#lang reader "../aoc-lang.rkt"
|
|
|
|
(provide (rename-out [#%mb #%module-begin]))
|
|
(define-macro (#%mb (STARS) (JMP) ...)
|
|
#`(#%module-begin
|
|
(time (escape (list->vector '(JMP ...)) 'STARS))))
|
|
|
|
(define (escape vec stars)
|
|
(let/ec exit
|
|
(for/fold ([pos 0])
|
|
([i (in-naturals)])
|
|
(unless (<= 0 pos (sub1 (vector-length vec)))
|
|
(exit i))
|
|
(define jmp (vector-ref vec pos))
|
|
(vector-set! vec pos ((if (and (eq? stars '★★) (>= jmp 3))
|
|
sub1
|
|
add1) jmp))
|
|
(+ pos jmp))))
|
|
|