diff options
Diffstat (limited to '09/2.rkt')
| -rw-r--r-- | 09/2.rkt | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/09/2.rkt b/09/2.rkt new file mode 100644 index 0000000..5c25482 --- /dev/null +++ b/09/2.rkt @@ -0,0 +1,26 @@ +#lang racket + +(require "../lib/utils.rkt") + +(define lines + (call-with-input-file "input" + (λ (fp) + (get-lines fp)))) + +(define (parse-data line) + (map string->number (string-split line))) + +(define data (map parse-data lines)) + +(define (diff lst) + (let loop ((cur lst) (ret '())) + (if (or (null? cur) (null? (cdr cur))) + (reverse ret) + (loop (cdr cur) (cons (- (cadr cur) (car cur)) ret))))) + +(define (predict lst) + (if (andmap (λ (x) (= x (car lst))) lst) + (car lst) + (- (car lst) (predict (diff lst))))) + +(apply + (map predict data)) |
