如何实现懒惰的“缩小地图”功能?

时间:2010-11-23 09:55:30

标签: recursion clojure map reduce cons

我正在尝试实现“缩小地图”功能。也就是说,它应该返回一个序列,该序列包含将f应用于coll的前两项的结果,然后是将f应用于该结果的结果,以及第三项的结果coll等等。

(def c [[0 0 0 0] [1 0 0 0] [0 1 0 0] [0 0 1 0] [0 0 0 1]])

(defn- sum-vector [v1 v2]
  (map + v1 v2))

(defn reduce-map [f coll & acc]
  (if (< (count coll) 2)
    (if (empty? acc) coll acc)
    (let [head (apply f (take 2 coll))
          tail (drop 2 coll)]
      (recur f (conj tail head) (conj acc head)))))

例如,调用此函数如下:

(reduce-map sum-vector c)

应该返回:

[[1 0 0 0] [1 1 0 0] [1 1 1 0] [1 1 1 1]]

(实际上,它应该可以返回第一个未经修改的项目,以更好地模仿map,但我可以稍后解决。)

是的,现在,这就是它的回报:

((1 1 1 1) (1 1 1 0) (1 1 0 0) (1 0 0 0))

如何在(ny)seq结束时“推”?

如果我将reduce-map替换为recur,这就是它返回的内容:

(((1 1 1 1) ((1 1 1 0) ((1 1 0 0) ((1 0 0 0))))))

上面的代码中recur和真递归有什么区别?

而且,是否有内置的,更好的,或更实用的方式来实施reduce-map

最后,我希望输出序列是懒惰的。我只是将整个事情包裹在lazy-seq吗?

1 个答案:

答案 0 :(得分:10)

这听起来有点像reductions

至于seq结尾处的“推”:一般来说,seqs没有“结束”,参见(iterate inc 0)

关于列表末尾的“推送”:列表不是为此而设计的。使用矢量。使用[],而不是nil为您的累加器播种。

至于lazy-seq:使用“true”递归而不是recur。这是一个例子:

(defn integer-seq
  [start]
  (lazy-seq
    (cons start (integer-seq (inc start)))))