(ns adventofcode2019.day01
    [:require [adventofcode2019.lib :refer :all]
              [clojure.string :as str]])

(defn get-fuel-req [mass]
  (reduce + (take-while #(> % 0) (drop 1 (iterate #(- (quot % 3) 2) mass)))))

(defn total-fuel-requirement-naive [input]
  (reduce + (map #(- (quot % 3) 2) input)))

(defn total-fuel-requirement [input]
  (reduce + (map get-fuel-req input)))

(defn day01 []
  (let [input (map parse-int (get-list-from-file (input-file)))] 
    (println (total-fuel-requirement-naive input))
    (println (total-fuel-requirement input))))