37 lines
1.2 KiB
Racket
37 lines
1.2 KiB
Racket
#lang racket
|
|
|
|
(require racket/set
|
|
"../lib.rkt")
|
|
|
|
(define (parse instr)
|
|
(match instr
|
|
[(regexp #rx"(.+) (.+)" (list _ op arg))
|
|
(cons (string->symbol op) (string->number arg))]))
|
|
|
|
(define input (list->vector (map parse (problem-input 8))))
|
|
|
|
(define (run instrs try)
|
|
(let loop ([pc 0] [acc 0] [ran (set)] [backtrack (const #f)])
|
|
(define (get-backtrack next-pc)
|
|
(if (and try (= pc try))
|
|
(const (loop next-pc acc (set-add ran pc) (const #f)))
|
|
backtrack))
|
|
(cond
|
|
[(>= pc (vector-length instrs)) acc]
|
|
[(set-member? ran pc) (if try (backtrack) acc)]
|
|
[else (match (vector-ref instrs pc)
|
|
[`(acc . ,arg) (loop (add1 pc) (+ acc arg) (set-add ran pc) backtrack)]
|
|
[`(nop . ,arg) (loop (add1 pc) acc (set-add ran pc) (get-backtrack (+ pc arg)))]
|
|
[`(jmp . ,arg) (loop (+ pc arg) acc (set-add ran pc) (get-backtrack (add1 pc)))])])))
|
|
|
|
(define (try-run instrs)
|
|
(for/or ([i (range (vector-length instrs))]
|
|
#:when (member (car (vector-ref instrs i)) '(nop jmp)))
|
|
(match-let* ([`(,op . ,arg) (vector-ref instrs i)])
|
|
(run instrs i))))
|
|
|
|
(define-values (part1 part2)
|
|
(values (run input #f) (try-run input)))
|
|
|
|
(show-solution part1 part2)
|