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/d12/main.rkt

30 lines
898 B
Racket

7 years ago
#lang reader "../aoc-lang.rkt"
(require graph)
(provide (rename-out [#%mb #%module-begin]))
(define-macro (#%mb (STARS) (NUM <-> . NUMS) ...)
#'(#%module-begin
7 years ago
(time
(define g (unweighted-graph/undirected null))
(for-each (curry add-edge! g NUM) (list . NUMS)) ...
(if (eq? 'STARS ')
(programs-in-group g 0)
(number-of-groups g (list NUM ...))))))
7 years ago
(define (programs-in-group g x) (length (group-of g x)))
(define (group-of g x)
(define-values (connects _) (dijkstra g x))
7 years ago
(for/list ([(k v) (in-mutable-hash connects)]
7 years ago
#:when (integer? v))
k))
(define (number-of-groups g nums)
(for/fold ([nums-seen null]
[group-count 0]
#:result group-count)
([num (in-list nums)]
#:unless (memv num nums-seen))
(values (append (group-of g num) nums-seen) (add1 group-count))))