clojure - 尝试 'group' 并展平 clojure 中嵌套结构的值

Question

又一个不安的夜晚试图弄清楚如何做到这一点。鉴于我已重新格式化嵌套数据结构以提高可读性以查找格式错误的声明:

(def tm (list 
          {:a 1 :b 2 :c 3 :child (list 
               {:a 4 :b 5 :c 6 :child (list 
                     {:a 40 :b 50 :c 60 :child (list {:a nil :b nil :c nil})})}                                                                                               
               {:a 70 :b 80 :c 90 :child (list {:a nil :b nil :c nil})})} 
          {:a 400 :b 500 :c 600 :child (list {:a nil :b nil :c nil})}
          {:a 7 :b 8 :c 9 :child (list 
               {:a 10 :b 11 :c 12 :child (list {:a nil :b nil :c nil})})})
  )

遍历可以在下面的dg123示例中看到

我得到以下信息:

=>(roll-down tm :a)
((1 4 40) (1 70) (7 10))

但期待的是:

((1 4 40) (1 70) (400) (7 10))

Answer 1

这是一个基于 zipper 的解决方案

(require '[clojure.zip :as z])

(defn root-to-leaf-paths [t]
   (loop [t t paths []] 
     (cond 
       (z/end? t) paths
       (z/branch? t) (recur (z/next t) paths) 
       :leaf (recur (z/next t) (conj paths (z/path t))))))

(defn roll-down [d kw]
   (->> (z/zipper :child :child nil {:child d})
        (root-to-leaf-paths)
        (map (comp rest (partial map kw)))))

注意:您的 tm 不是一棵树，而是一个树列表。因此，在这里我们首先通过将树列表放置在一个父根 {:child d} 下，将此结构转换为一棵树。然后我们将其从结果路径中删除(comp rest ...)。

示例

(roll-down tm :a)
;=> ((1 4 40) (1 70) (400) (7 10))

(roll-down tm :b)
;=> ((2 5 50) (2 80) (500) (8 11))

(roll-down tm :c)
;=> ((3 6 60) (3 90) (600) (9 12))