tarcieri-4clojure-solution28.clj

;; tarcieri's solution to Flatten a Sequence
;; https://4clojure.com/problem/28

(fn zomgflatten
  ([input] 
    (seq (zomgflatten input [])))
  ([initial-seq initial-result]
    (loop [s initial-seq result initial-result]
      (let [head (first s) tail (rest s)]
        (if head
          (if (or (list? head) (vector? head))
            (let [res (zomgflatten head result)]
              (recur tail res))
            (let [res (conj result head)]
              (recur tail res)))
          result)))))