From 16f101de883c30af39e791d71506d2311bc38e08 Mon Sep 17 00:00:00 2001 From: Matthew Butterick Date: Thu, 1 Dec 2016 14:34:17 -0800 Subject: [PATCH] steps --- 2016/day01.rkt | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/2016/day01.rkt b/2016/day01.rkt index 999d930..1c16f80 100644 --- a/2016/day01.rkt +++ b/2016/day01.rkt @@ -22,7 +22,14 @@ (+ (abs (imag-part loc)) (abs (real-part loc)))) (define (locs-between loca locb) - (range loca (add1 locb))) + (define real-steps (abs (- (real-part loca) (real-part locb)))) + (define imag-steps (abs (- (imag-part loca) (imag-part locb)))) + (define diff (- locb loca)) + (let loop ([locs (list loca)]) + (if (equal? (car locs) locb) + (reverse locs) + (cons (+ (real-part loca) + (* +i (imag-part loca))) locs)))) (module+ test (require rackunit)