aboutsummaryrefslogtreecommitdiff
path: root/2020/day-01/day-01.rkt
blob: 56c0b87c65cdab7e9e72e512fc8785fc2ef1b392 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#lang racket
(require "../../jj-aoc.rkt"
         threading)

(define entries
  (~>> (open-day 01 2020)
       (port->list read)
       list->set))

;; part 1
(define (look-for-complement xs)
  (define x (set-first xs))
  (cond
    [(set-member? xs (- 2020 x)) (* x (- 2020 x))]
    [else (look-for-complement (set-rest xs))]))

(time (look-for-complement entries))

;; part 2
(time (for*/first ([x (in-set entries)]
                   [y (in-set (set-subtract entries (set x)))]
                   #:when (set-member? (set-subtract entries (set x y)) (- 2020 x y)))
        (* x y (- 2020 x y))))