clojure 斐波那契数列一个解法的解析

最近做 4Clojure的fib序列的一个题 http://www.4clojure.com/problem/26

自己用的是

#(take % (map first (iterate (fn [[a b]] [b (+ a b)]) [1 1])))

做完以后看到一个答案用的是这种方式

#((apply comp (repeat (- % 2) (fn [x] (conj x (+ (peek x) (peek (pop x))))))) [1 1])

看了半天才理解下来是怎么回事，这里记录下来


这里把代码拆开来 类似于

(def compute (fn [x] (conj x (+ (peek x) (peek (pop x))))))
(def fn-seq (repeat (- n 2) compute)
(def fib (apply comp fn-seq))
(fib [1 1])

首先

(fn [x] (conj x (+ (peek x) (peek (pop x)))))

这段函数单独来看比较好理解，就是把vector尾部的两个记录相加，添加到x的末尾

(repeat (- % 2) (fn [x] (conj x (+ (peek x) (peek (pop x))))))

重复生成了(- % 2) 次 后面定义的那个函数，返回一个包含同一个fn的seq

(apply comp fn-seq)

就是把fn-seq作为参数，调用comp，来生成新的fn，这个fn的作用是不停的调用上面定义的那个fn compute

最后是调用fn,传入参数[1 1]
当 n=3的时候
就会(compute [1 1]) => [1 1 2]
n=4 时
(compute (compute [1 1])) => [1 1 2 3]

不过这个解法还是有bug
n<=1的时候，还是返回[1 1]
repeat反回了空seq ()

看了comp的源代码，参数为空时会返回identity
这样就直接返回了[1 1]