src/problems/fibd.lisp @ b5923704ce42

(in-package :rosalind)

(define-problem fibd (data stream)
    "6 3"
    "4"
  (iter
    (with months = (read data))
    (with lifespan = (read data))
    (for month :from 2 :to months)
    (labels ((ref (array index)
               (if (plusp index)
                 (aref array index)
                 0))
             (breeding (month)
               (- (ref population (- month 1))
                  (deaths month)))
             (births (month)
               (breeding (- month 1)))
             (deaths (month)
               (ref births (- month lifespan)))
             (population (month)
               (+ (breeding month)
                  (births month))))
      ;; We initialize the buffers with NIL in index 0 for the 1-based months,
      ;; and 1 in index 0 for the initial pair of rabbits.
      (buffering (returning-final (population month))
                 :into population
                 :initial-contents '(nil 1))
      (buffering (births month)
                 :into births
                 :initial-contents '(nil 1)))))

(problem-fibd "45 6")
;; (solve fibd)