aboutsummaryrefslogtreecommitdiff
path: root/racket/leetcode/lc-1018-binary-prefix.rkt
blob: fa826812ed18e7a5e9f59e2c1071054a40b87cf5 (plain)
1
2
3
4
5
6
7
8
9
10
11
#lang racket
(define/contract (prefixes-div-by5 A)
  (-> (listof exact-integer?) (listof boolean?))
  (define ns (make-vector (length A) #false))
  (for/fold ([acc 0])
            ([b (in-list A)]
             [i (in-naturals)])
    (let ([test-val (remainder (+ (* 2. acc) b) 5)])
      (when (= 0 test-val) (vector-set! ns i #true))
      test-val))
  (vector->list ns))