折叠在玫瑰树上

Question

给定以下Algrebaic数据结构：

data Tree a = Node {
    rootLabel :: a,
    subForest :: [Tree a]
}  deriving (Show)

并fold ：

treeFold :: (a -> [b] -> b) -> Tree a -> b
treeFold f (Node x ts) = f x (map (treeFold' f) ts)

我如何从Tree a获得[a] ？

Answer 1

你是说用折纸吗？ 您可以得到一个函数Tree a -> [a]非常简单：

collapse :: Tree a -> [a]
collapse (Node x ts) = x : (concat $ map collapse ts)

Prelude> let t = Node 3 [Node 2 [], Node 4 [], Node 6 []]
Prelude> collapse t
[3,2,4,6]

如果您特别想使用fold ，我想您可以做类似的事情：

collapse' :: Tree a -> [a]
collapse' = treeFold (\x tss -> x : (concat tss))

Prelude> collapse' t
[3,2,4,6]

我个人认为第一个版本更清晰。