clojure - 更改 Clojure 中的映射行为

Question

我需要修改映射函数行为，以提供不具有最小集合大小但具有最大集合大小的映射，并使用零来表示丢失的元素。

标准行为:

(map + [1 2 3] [4 5 6 7 8]) => [5 7 9]

所需行为:

(map + [1 2 3] [4 5 6 7 8]) => [5 7 9 7 8]

我编写了函数来执行此操作，但它似乎不能很好地使用可变参数进行扩展。

(defn map-ext [f coll1 coll2]
  (let [mx (max (count coll1) (count coll2))]
    (map f
     (concat coll1 (repeat (- mx (count coll1)) 0))
     (concat coll2 (repeat (- mx (count coll2)) 0)))))

有更好的方法吗？

Answer 1

你的方法很简洁，但效率低下(它调用count)。一种更有效的解决方案，不需要将其整个输入序列存储在内存中:

(defn map-pad [f pad & colls]
  (lazy-seq
   (let [seqs (map seq colls)]
     (when (some identity seqs)
       (cons (apply f (map #(or (first %) pad) seqs))
             (apply map-pad f pad (map rest seqs)))))))

像这样使用:

user=> (map-pad + 0 [] [1] [1 1] (range 1 10))
(3 3 3 4 5 6 7 8 9)

编辑:将map-pad广义化为任意数量。