Question

这段代码来自我已经解决的作业。尽管如此，我仍在尝试找出是否可以解决最初的尝试。

所以我们有了这个树结构和foldTree函数。

data Tree a = Leaf a
 | Node (Tree a) (Tree a)

foldTree :: (b -> b -> b) -> (a -> b) ->  Tree a -> b
foldTree op f (Leaf x)  = f x
foldTree op f (Node l r) = foldTree op f l `op` foldTree op f r

现在必须使用foldTree实现mapTree。我是这样完成的。

 mapTree :: (a -> b) -> Tree a -> Tree b
 mapTree' f tree = foldTree Node (Leaf . f) tree

我最初想到但仍然无法工作的是这样的：

 mapTree :: (a -> b) -> Tree a -> Tree b
 mapTree f tree = foldTree Node transFunc tree 
   where transFunc :: Tree a -> Tree b
         transFunc (Leaf x)    = Leaf (f x)
         transFunc (Node l r)  = Node (transFunc l) (transFunc r)

Answer 1

第二个函数是错误的，因为它的类型为：Tree a -> Tree b，而foldTree期望它是a -> b，其中a来自Tree a。 b要求mapTree为Tree b，因此foldTree的第三个参数应为a -> Tree b类型。

因此，transFunc的最简单的固定版本是：

mapTree :: forall a b. (a -> b) -> Tree a -> Tree b
mapTree f tree = foldTree Node transFunc tree 
  where transFunc :: a -> Tree b
        transFunc x = Leaf (f x)

请注意，您需要启用ScopedTypeVariables扩展名才能对其进行编译。

transFunc的该版本等效于您的工作解决方案：(Leaf . f)

Haskell mapTree使用foldTree实现

1 个答案: