列出树中叶子的路径

问题描述 投票:4回答:3

我正在尝试编写一个函数来查找树中叶子的所有路径。例如,给定一个如下所示的树:

           1
         /  \
        2    5
       / \    \
      3   4    6

输出列表将是:[[1,2,3],[1,2,4],[1,5,6]]

此函数的类型签名是:branches :: Tree a -> [[a]]。请注意,这使用了Data.Tree包中定义的Tree类型,虽然示例树是二进制的,但实际的树类型是玫瑰树。

list haskell tree
3个回答
1
投票

我会通过在部分路径列表的前面添加新标签来构建路径,然后将它们反转以进行输出:

listtree tree = map reverse $ traverse [] tree
  where traverse path (Node label []) = [label:path]
        traverse path (Node label xs) = concat $ map (traverse (label:path)) xs

将标签添加到列表前面而不是结尾的原因是追加时间为O(N)并分配内存,而添加头部则需要O(1)。当然,反转列表也是O(N),但每个列表只进行一次反转...

因此,在使用功能算法和数据结构时,上述“添加到头部,然后在必要时反转”模式是一种普遍的习惯用法。


编辑:从@ luqui的评论中,获取路径的一种补充方法是从下到上构建它们:

listtree (Node label []) = [[label]]
listtree (Node label xs) = map (label:) $ concat $ map listtree xs

这比我的解决方案更短(并且可能更清晰),并且它还具有以您希望的顺序为您提供路径的额外优势:路径是从叶子而不是从根开始构建的。

注意(与前面的解决方案一样)路径列表通过在列表的开头添加一个头而不是追加到末尾来扩展。


3
投票

只是一个简单的功能:

listtree :: [a] -> Tree a -> [[a]]
listtree l (Node a []) = [l++[a]]
listtree l (Node a forest) = concatMap (listtree (l++[a])) forest

使用列表记录从根节点到当前节点的路径,并将当前节点的标签附加到路径,然后递归地将listtree映射到每个子节点。

listtree [] (Node 1 [(Node 2 [(Node 3 []), (Node 4 [])]), (Node 5 [(Node 6 [])])]))

产生理想的结果[[1,2,3],[1,2,4],[1,5,6]]


0
投票

这很简单......这些是你需要考虑的事情: -

  1. 递归调用该函数
  2. 在函数内,遍历应该是PreOrder(root,left和right)的形式
  3. 在递归函数中使用数组并存储所访问的每个节点的值。
  4. 如果访问节点是叶子...节点 - >左== NULL和节点 - >右== NULL。然后输出数组的内容。
© www.soinside.com 2019 - 2024. All rights reserved.