Clojure的地圖轉換（孩子父親在樹上）

Question

說我有一個樹和節點孩子的以下地圖：

(def a {0 [], 1 [], 2 [0, 1]}) 

在其根部和兩個對應於樹節點葉節點和作爲節點的孩子。

我如何將它轉換成父親的地圖，或者更好的是，將它與父親一起裝飾。例如。在父親的下列地圖到達：

{0 2, 1 2, 2 nil} ; each node only has one father at most 

或者，更好的是，在下面的圖，結合孩子和父親：

{0 [[] 2], 1 [[] 2], 2 [[0,1] nil]} 

Answer 1

(defn parents [m] 
    (let [plist (into {} (for [[k v] m vv v] [vv k]))] 
    (into {} (map (fn [[k v]] [k [v (plist k)]]) m)))) 

(parents a) 
=> {0 [[] 2], 1 [[] 2], 2 [[0 1] nil]} 

Answer 2

第一個位：

(def a {0 [], 1 [], 2 [0, 1]}) 

(defn parent-map [m] 
    (reduce 
    (fn [x [k v]] 
     (into x (zipmap v (repeat k)))) {} m)) 

(def parent (parent-map a)) 

parent 
=> {1 2, 0 2} 
(parent 1) 
=> 2 
(parent 2) 
=> nil 

所以，不需要明確在父映射中有2 nil。

二位：

(defn parent-child-map [m] 
    (let [parent (parent-map m)] 
    (reduce 
     (fn [x [k v]] 
     (assoc x k [(m k) (parent k)])) {} m))) 

(parent-child-map a) 
=> {2 [[0 1] nil], 1 [[] 2], 0 [[] 2]} 

的東西一點點更有趣：

(def b {0 [], 1 [], 2 [], 3 [], 4 [0 1 2], 5 [3], 6 [4 5]}) 

(parent-child-map b) 
=> 
{6 [[4 5] nil], 
5 [[3] 6], 
4 [[0 1 2] 6], 
3 [[] 5], 
2 [[] 4], 
1 [[] 4], 
0 [[] 4]}