我正在尝试实现一个递归函数,它接受一个树并列出它拥有的所有路径。
我当前的实施不起作用:
let rec pathToList tree acc =
match tree with
| Leaf -> acc
| Node(leftTree, x, rightTree) ->
let newPath = x::acc
pathToList leftTree newPath :: (pathToList rightTree newPath)
...因为pathToList leftTree newPath
不返回元素而是返回列表,因此错误。
这可以用set修复,例如:
let rec pathToList2 t acc =
match t with
| Leaf -> Set.singleton acc
| Node(leftTree, x, rightTree) ->
let newPath = x::acc
Set.union (pathToList2 leftTree newPath) (pathToList2 rightTree newPath)
现在,我有点坚持使用列表(前)而不是设置(稍后)来解决这个问题,关于如何使用尾递归解决这个问题的任何建议?
答案 0 :(得分:2)
为了解决这个问题而不诉诸@
(因为它在第一个列表的长度上是线性的,效率很低),你需要两个累加器:一个用于到父节点的路径(所以您可以构建当前节点的路径),并为目前为止找到的所有路径建立路径(这样您就可以添加找到的路径)。
let rec pathToListRec tree pathToParent pathsFound =
match tree with
| Leaf -> pathsFound
| Node (left, x, right) ->
let pathToHere = x :: pathToParent
// Add the paths to nodes in the right subtree
let pathsFound' = pathToListRec right pathToHere pathsFound
// Add the path to the current node
let pathsFound'' = pathToHere :: pathsFound'
// Add the paths to nodes in the left subtree, and return
pathToListRec left pathToHere pathsFound''
let pathToList1 tree = pathToListRec tree [] []
就尾递归而言,您可以看到上述函数中的两个递归调用之一位于尾部位置。但是,仍有一个非尾部位置的呼叫。
这是树处理函数的经验法则:你不能轻松地使它们完全是尾递归的。原因很简单:如果你天真地做,两个递归中的至少一个(向下到左子树或到右子树)必然处于非尾部位置。唯一的方法是使用列表模拟调用堆栈。这意味着您将具有与非尾递归版本相同的运行时复杂性,除非您使用列表而不是系统提供的调用堆栈,因此它可能会更慢。
无论如何它是这样的:
let rec pathToListRec stack acc =
match stack with
| [] -> acc
| (pathToParent, tree) :: restStack ->
match tree with
| Leaf -> pathToListRec restStack acc
| Node (left, x, right) ->
let pathToHere = x :: pathToParent
// Push both subtrees to the stack
let newStack = (pathToHere, left) :: (pathToHere, right) :: restStack
// Add the current path to the result, and keep processing the stack
pathToListRec newStack (pathToHere :: acc)
// The initial stack just contains the initial tree
let pathToList2 tree = pathToListRec [[], tree] []
代码看起来并不太糟糕,但它需要的时间是非尾递归的两倍,并且会进行更多的分配,因为我们使用列表来完成堆栈的工作!
> #time;;
--> Timing now on
> for i = 0 to 100000000 do ignore (pathToList1 t);;
Real: 00:00:09.002, CPU: 00:00:09.016, GC gen0: 3815, gen1: 1, gen2: 0
val it : unit = ()
> for i = 0 to 100000000 do ignore (pathToList2 t);;
Real: 00:00:21.882, CPU: 00:00:21.871, GC gen0: 12208, gen1: 3, gen2: 1
val it : unit = ()
总结:规则“让它尾随递归会更快!”当需要进行多次递归调用时,不应该遵循极限,因为它需要以一种使其更慢的方式更改代码。