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.
33 lines
973 B
Racket
33 lines
973 B
Racket
6 years ago
|
#lang debug br
|
||
|
|
|
||
|
|
(define ps "08.txt")
|
||
|
|
|
||
|
|
(define (★)
|
||
|
|
(define (node-value)
|
||
|
|
(define child-count (read))
|
||
|
|
(define metadata-count (read))
|
||
|
|
(+ (for/sum ([c (in-range child-count)])
|
||
|
|
(node-value))
|
||
|
|
(for/sum ([c (in-range metadata-count)])
|
||
|
|
(read))))
|
||
|
|
(with-input-from-file ps node-value))
|
||
|
|
|
||
|
|
(define (★★)
|
||
|
|
(define (node-value)
|
||
|
|
(define child-count (read))
|
||
|
|
(define metadata-count (read))
|
||
|
|
(define child-values (for/list ([c (in-range child-count)])
|
||
|
|
(node-value)))
|
||
|
|
(define metadatas (for/list ([c (in-range metadata-count)])
|
||
|
|
(read)))
|
||
|
|
(if (zero? child-count)
|
||
|
|
(apply + metadatas)
|
||
|
|
(for/sum ([md (in-list metadatas)]
|
||
|
|
#:when (<= 1 md child-count))
|
||
|
|
(list-ref child-values (sub1 md)))))
|
||
|
|
(with-input-from-file ps node-value))
|
||
|
|
|
||
|
|
(module+ test
|
||
|
|
(require rackunit)
|
||
|
|
(check-equal? (time (★)) 44838)
|
||
|
|
(check-equal? (time (★★)) 22198))
|