[4Clojure]解题记录-#69

Difficulty:	Medium
Topics:	core-functions

 

Write a function which takes a function f and a variable number of maps. Your function should return a map that consists of the rest of the maps conj-ed onto the first. If a key occurs in more than one map, the mapping(s) from the latter (left-to-right) should be combined with the mapping in the result by calling (f val-in-result val-in-latter)

 

(= (__ * {:a 2, :b 3, :c 4} {:a 2} {:b 2} {:c 5})  {:a 4, :b 6, :c 20})

 

(= (__ - {1 10, 2 20} {1 3, 2 10, 3 15}) {1 7, 2 10, 3 15})

 

(= (__ concat {:a [3], :b [6]} {:a [4 5], :c [8 9]} {:b [7]}) {:a [3 4 5], :b [6 7], :c [8 9]})

 

特殊限制：禁用merge-with

 

解（长度：100）

(fn [f & x] (reduce 
  (fn [m v]
  (let [k (first v) w (last v)]
    (if (m k)
      (assoc m k (f (m k) w))
      (assoc m k w)))) {} (apply concat x)))

中间过程：经历了2个版本，第一个版本比较差，后面这个比较好（短）。
基本思路是用reduce两两合并，输入方式是(apply concat x)，效果：
(apply concat '({:a 2, :b 3, :c 4} {:a 2} {:b 2} {:c 5}))
；([:a 2] [:b 3] [:c 4] [:a 2] [:b 2] [:c 5])
因此第一个reduce合并的想法是，vector of vector 与 vector 合并：
(defn my-com [f x y]
  (let [u (last x) v (first u) w (first y)]
    (if (= v w)
      (sort-by  #(first %) (conj (butlast x) [v (f (last u) (last y))]))
      (sort-by  #(first %) (conj x y)))))
效果：（这里需要有序的输入）
user=> (my-com + [] [:a 1]);([:a 1]) 
user=> (my-com + [[:a 3] [:b 4]] [:b 1]);([:a 3] [:b 5])

最后配合reduce：(reduce (partial my-com +) nil (sort-by #(first %) (concat {:a 2:b 3} {:a 2 :b 3 :c 4}))); ([:a 4] [:b 6] [:c 4])； 以及按要求输出成{:a 4, :b 3, :c 4}，
(into (hash-map) (reduce (partial my-com *) nil (sort-by #(first %) (concat{:a 2, :b 3, :c 4} {:a 2} {:b 2} {:c 5}))));{:a 4, :b 6, :c 20}

然后是所有函数匿名以及整合。 但是这个方法不好，因为中间通过vector的形式合并，又需要有序，因此立刻想到reduce是hash-map与vector的合并：
(defn my-com2 [f m v]
  (let [k (first v) w (second v)]
    (if (m k)
      (assoc m k (f (m k) w))
      (assoc m k w))))
user=>(my-com2 + {:a 2 :b 3} [:a 1]);{:a 3, :b 3}
user=>(reduce (partial my-com2 *) {} (apply concat '({:a 2, :b 3, :c 4} {:a2} {:b 2} {:c 5}))); {:c 20, :b 6, :a 4}

这次，就不需要对输出进行转换了，因为直接是目标格式的大map，最终整合，匿名时候发现，my-com2的匿名形式不需要f这个参数，因为在外层的fn[f & x]已经有了f，可以直接用：

(fn [f & x] (reduce 
  (fn [m v]
  (let [k (first v) w (last v)]
    (if (m k)
      (assoc m k (f (m k) w))
      (assoc m k w)))) {} (apply concat x)))

为了尽可能缩短代码，把second 换成了last，这个纯粹是为了题目，其实不是好的做法。

转载于:https://www.cnblogs.com/zjfu/p/4124736.html