From 90d1dfa982e02ac4aae5f507a7bdae4671855e26 Mon Sep 17 00:00:00 2001
From: Jonathan Chan <jcxz@cs.ubc.ca>
Date: Wed, 8 Dec 2021 23:17:22 -0800
Subject: [PATCH] Day 09: Use sets.

---
 src/09.rkt | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/src/09.rkt b/src/09.rkt
index 5c91526..0ca2006 100644
--- a/src/09.rkt
+++ b/src/09.rkt
@@ -1,6 +1,7 @@
 #lang curly-fn racket
 
-(require racket/set "../lib.rkt")
+(require racket/set
+         "../lib.rkt")
 
 (define input
   (~>> (problem-input 9)
@@ -27,18 +28,20 @@
               (grid-ref input (car %) (cdr %))}
           (neighbours row col)))
 
-(define (highers row col)
-  (filter #{and (< (grid-ref input row col)
-                   (grid-ref input (car %) (cdr %) -inf.0)
-                   9)}
-          (neighbours row col)))
+(define (basinic seen loc)
+  (list->set
+   (filter #{and (< (grid-ref input (car %) (cdr %)) 9)
+                 (not (set-member? seen %))}
+           (neighbours (car loc) (cdr loc)))))
 
-(define (basin unseen seen)
-  (match unseen
-    ['() (unique seen)]
-    [(list (and hd (cons row col)) tl ...)
-     (basin (append tl (highers row col))
-            (cons hd seen))]))
+(define (basin row col)
+  (let loop ([unseen (set (cons row col))]
+             [seen (set)])
+    (if (set-empty? unseen)
+        (set-count seen)
+        (let ([loc (set-first unseen)])
+          (loop (set-union (basinic seen loc) (set-rest unseen))
+                (set-add seen loc))))))
 
 (define-values (part1 part2)
   (for/fold ([risk 0]
@@ -50,7 +53,7 @@
               ([col (range cols)])
       (if (low? row col)
           (values (+ (add1 (grid-ref input row col)) risk)
-                  (cons (length (basin `((,row . ,col)) '())) basins))
+                  (cons (basin row col) basins))
           (values risk basins)))))
 
 (show-solution part1 part2)
\ No newline at end of file