tree map

2021-12-17 07:23:03 +11:00
parent bf2cae7c4e
commit d9940d81d0
3 changed files with 12 additions and 4 deletions
--- a/lib/tree.ml
+++ b/lib/tree.ml
@@ -1,5 +1,10 @@
-open List
+open FromStdlib
 open Types

 let combine (tr1 : 'a tree) (tr2 : 'a tree) (topBranch : 'a) : 'a tree =
-    Branch (topBranch, tr1 :: tr2 :: [])
+    Branch (topBranch, tr1 :: tr2 :: [])
+
+let rec map (f : 'a -> 'b) (tr : 'a tree) : 'b tree =
+    match tr with
+    | Branch (value, branches) -> Branch (f value, branches |> List.map (map f))
+    | Leaf -> Leaf