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.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.
aoc-racket/2017/d1/main.rkt

13 lines
471 B
Racket

#lang reader "../aoc-lang.rkt"
(provide (rename-out [#%mb #%module-begin]))
(define-macro (#%mb (STARS) (NUMBER) ...)
#'(#%module-begin (captcha-sum 'STARS NUMBER) ...))
(define (captcha-sum stars num)
(define digits (number->digits num))
(define offset (if (eq? stars ') -1 (quotient (length digits) 2)))
(for/sum ([digit (in-list digits)]
[other-digit (in-list (shift-cycle digits offset))]
#:when (= digit other-digit))
digit))