我正在尝试使Brian's Fold for Bianary Trees(http://lorgonblog.wordpress.com/2008/04/06/catamorphisms-part-two/)适应申请Multiway树。
总结Brian的博客:
数据结构:
type Tree<'a> =
| Node of (*data*)'a * (*left*)Tree<'a> * (*right*)Tree<'a>
| Leaf
let tree7 = Node(4, Node(2, Node(1, Leaf, Leaf), Node(3, Leaf, Leaf)),
Node(6, Node(5, Leaf, Leaf), Node(7, Leaf, Leaf)))
二叉树折叠功能
let FoldTree nodeF leafV tree =
let rec Loop t cont =
match t with
| Node(x,left,right) -> Loop left (fun lacc ->
Loop right (fun racc ->
cont (nodeF x lacc racc)))
| Leaf -> cont leafV
Loop tree (fun x -> x)
实施例
let SumNodes = FoldTree (fun x l r -> x + l + r) 0 tree7
let Tree6to0 = FoldTree (fun x l r -> Node((if x=6 then 0 else x), l, r)) Leaf tree7
多线树版本 [未(完全)正常工作] :
数据结构
type MultiTree = | MNode of int * list<MultiTree>
let Mtree7 = MNode(4, [MNode(2, [MNode(1,[]); MNode(3, [])]);
MNode(6, [MNode(5, []); MNode(7, [])])])
折叠功能
let MFoldTree nodeF leafV tree =
let rec Loop tree cont =
match tree with
| MNode(x,sub)::tail -> Loop (sub@tail) (fun acc -> cont(nodeF x acc))
| [] -> cont leafV
Loop [tree] (fun x -> x)
示例1 返回28 - 似乎有效
let MSumNodes = MFoldTree (fun x acc -> x + acc) 0 Mtree7
示例2
不运行
let MTree6to0 = MFoldTree (fun x acc -> MNode((if x=6 then 0 else x), [acc])) Mtree7
最初我认为MFoldTree
在某个地方需要map.something
,但我让它与@
运营商合作。
第二个例子的任何帮助和/或纠正我在MFoldTree
函数中所做的事情都会很棒!
干杯
dusiod
答案 0 :(得分:9)
诀窍是你需要传递一个额外的功能来折叠。
在Brian的版本中,fold函数只需要使用节点中的值调用nodeF
,并从左侧和右侧子树生成两个值。
这对多路树来说还不够。在这里,我们需要一个函数nodeF
,它使用节点中的值调用,并通过聚合子树的所有值生成结果。但是你还需要一个函数 - 比如combineF
,它结合了节点的多个子树产生的值。
您的折叠功能是一个良好的开端 - 您只需要再添加一个递归调用来处理tail
:
let MFoldTree nodeF combineF leafV tree =
let rec Loop trees cont =
match trees with
| MNode(x,sub)::tail ->
// First, process the sub-trees of the current node and get
// a single value called 'accSub' representing (aggregated)
// folding of the sub-trees.
Loop sub (fun accSub ->
// Now we can call 'nodeF' on the current value & folded sub-tree
let resNode = nodeF x accSub
// But now we also need to fold all remaining trees that were
// passed to us in the parameter 'trees'..
Loop tail (fun accTail ->
// This produces a value 'accTail' and now we need to combine the
// result from the tail with the one for the first node
// (which is where we need 'combineF')
cont(combineF resNode accTail) ))
| [] -> cont leafV
Loop [tree] (fun x -> x)
求和很简单,因为我们只使用+
运算符来表示这两个函数:
let MSumNodes = MFoldTree (+) (+) 0 Mtree7
过滤树更加棘手。 nodeF
函数将获取节点中的元素和子节点的列表(这是聚合的结果)并生成单个节点。 combineF
函数将从第一个节点(即MultiTree
值)获取结果,并从其余节点生成子项列表。从空树生成的初始值是一个空列表:
let MTree6to0 =
MFoldTree (fun x children -> MNode((if x=6 then 0 else x), children))
(fun head tail -> head::tail) [] Mtree7
答案 1 :(得分:5)
另一种解决方案可能是
let rec mfold f a (MNode(x,s)) = f (List.fold (fun a t -> mfold f a t) a s) x
实际上,我们可以将树视为一个直线结构(折叠它)。
用例
> mfold (+) 0 Mtree7;;
val it : int = 28
过滤与正常折叠相同(因为mfold
是正常折叠):
> mfold (fun a x -> if x = 6 then a else x + a) 0 Mtree7;;
val it : int = 22
该功能可以是通用的(如List.fold
,Array.fold
,......可以是泛型。)
“但第二个意图是返回修改的整个树,以便任何具有值6的节点现在具有值0”
但这不是fold
计算,而是map
!
你可以做easyilly(再次作为一个线性结构处理)
let rec mmap f (MNode(x,s)) = MNode(f x, List.map (mmap f) s)
用例
> mmap (fun x -> if x=6 then 0 else x) Mtree7;;
val it : MultiTree =
MNode
(4,
[MNode (2,[MNode (1,[]); MNode (3,[])]);
MNode (0,[MNode (5,[]); MNode (7,[])])])
同样,我建议为每个可能的列表容器(Seq
,List
,Array
,...)执行此操作,它允许用户在上下文中选择最佳策略。< / p>
注意: