在树上的haskell折叠操作

时间:2013-04-26 01:29:24

标签: haskell tree fold

data Tree a = Tree a [Tree a]

请注意,我们不允许使用空树,并且叶子是带有空子树列表的树。

treeFold :: (a -> [b] -> b) -> Tree a -> b
treeFold f (Tree x s) = f x (map (treeFold f) s)

鉴于以上信息,我不明白树折功能如何 通过递归地将折叠操作应用于子树,然后将函数应用于根处的标签以及从子树返回的结果来返回结果。

我也不知道树折叠函数如何只接受一个参数而不是2,当它作为参数传递给map函数时它仍然可以编译并正常运行。

例如,下面的树大小函数,计算树的节点。

treeSize :: Tree a -> Int
treeSize = treeFold (\x ys -> 1 + sum ys)

运行 treeSize树,其中tree = Tree 4 [Tree 1 [Tree 2 [], Tree 3 []]]将树的大小设为4。

在上面的树大小函数中,树折函数也传递一个参数而不是两个。另外,传递给树折函数的x并没有在任何地方使用,所以为什么你需要它。删除它会导致程序无法编译,并提供以下错误消息。

 Couldn't match type `a' with `[[Int] -> Int]'
      `a' is a rigid type variable bound by
          the type signature for treeSize :: Tree a -> Int
          at treeFold.hs:15:1
    In the first argument of `sum', namely `ys'
    In the second argument of `(+)', namely `sum ys'
    In the expression: 1 + sum ys

非常感谢任何帮助。

2 个答案:

答案 0 :(得分:9)

未使用的参数

首先,将变量明确标记为未使用的方式是将变量替换为_。所以你真的想要:

treeFold (\_ ys -> 1 + sum ys)

您收到编译错误,因为您写了:

treeFold (\ys -> 1 + sum ys)

......这不是一回事。

褶皱

其次,我会在示例树上评估treeSize,以便您可以看到没有魔法:

treeSize (Tree 1 [Tree 2 [], Tree 3 []])

-- Inline definition of 'treeSize'
= treeFold (\_ ys -> 1 + sum ys) (Tree 1 [Tree 2 [], Tree 3 []])

-- Evaluate treeFold
= (\_ ys -> 1 + sum ys) 1 (map (treeFold (\_ ys -> 1 + sum ys)) [Tree 2 [], Tree 3 []])

-- Apply the anonymous function
= 1 + sum (map (treeFold (\_ ys -> 1 + sum ys)) [Tree 2 [], Tree 3 []])

-- Apply the 'map' function
= 1 + sum [ treeFold (\_ ys -> 1 + sum ys) (Tree 2 [])
          , treeFold (\_ ys -> 1 + sum ys) (Tree 3 [])
          ]

-- Apply both 'treeFold' functions
= 1 + sum [ (\_ ys -> 1 + sum ys) 2 (map (treeFold (\_ ys -> 1 + sum ys)) [])
          , (\_ ys -> 1 + sum ys) 3 (map (treeFold (\_ ys -> 1 + sum ys)) [])
          ]

-- Apply the anonymous functions
= 1 + sum [ 1 + sum (map (treeFold (\_ ys -> 1 + sum ys)) [])
          , 1 + sum (map (treeFold (\_ ys -> 1 + sum ys)) [])
          ]

-- map f [] = []
= 1 + sum [ 1 + sum []
          , 1 + sum []
          ]

-- sum [] = 0
= 1 + sum [1 + 0, 1 + 0]
= 1 + sum [1, 1]

-- Apply 'sum'
= 1 + 2
= 3

但是,有一种简单的方法可以记住treeFold的工作原理。它只是用你提供的函数替换每个Tree构造函数。

所以如果你有:

Tree 1 [Tree 2 [Tree 3 [], Tree 4[]], Tree 5 [Tree 6 [], Tree 7 []]]

...并且您将treeFold f应用于此,它将评估为:

f 1 [f 2 [f 3 [], f 4 []], f 5 [f 6 [], f 7 []]]

treeSum只是f = (\_ ys -> 1 + sum ys)

的特例
1 + sum [1 + sum [1 + sum [], 1 + sum []], 1 + sum [1 + sum [], 1 + sum []]]

= 7

柯里

最后一点是在Haskell中如何运作。定义如下函数时:

foo x y = x + y

...编译器实际上将它解释为两个嵌套函数:

foo = \x -> (\y -> x + y)

这就是为什么你可以将函数部分应用于Haskell中的一个参数。当您撰写foo 1时,它会转换为:

foo 1

= (\x -> (\y -> x + y)) 1

= \y -> 1 + y

换句话说,它返回一个参数的新函数。

在Haskell中,所有函数只接受一个参数,并通过返回新函数来模拟多个参数的函数。这种技术被称为“currying”。

如果您更喜欢更传统主流语言的多参数方法,可以通过让函数接受元组参数来模拟它:

f (x, y) = x + y

然而,这不是真正惯用的Haskell,它不会给你任何性能提升。

答案 1 :(得分:1)

第一个问题有点棘手,因为这是递归的事情......正如教师所说:“要理解递归,你必须学习,递归是如何工作的”。 :-P 一个小建议:尝试使用单个树或一棵树内的一个树的treeFold的应用程序,并由您自己(在纸上左右)评估它。我想,那么你可以了解发生了什么......(当然使用一个简单的函数作为treeFold的参数;就像你的treeSize使用的那样)。

treeFold只在地图正文中获得一个参数,因为map需要a->b的函数,treeFold的类型为(a -> [b] -> b) -> Tree a -> b.,所以如果你会传递2个参数,你只会传递给map一个值,这会导致失败。 (一个可以理解的例子:(+)需要两个参数。如果你写map (+1) [1,2,3],你将获得[2,3,4] ,因为(+1)应用于列表中的每个元素(并且(+ 1)显然需要一个参数,如上面的treeFold f

您的示例treeSize: 当你说,它只有一个参数时,这是不对的。你可以写

treeSize t = treeFold (\x ys -> 1 + sum ys) t

而不是上面的定义。 x没有被使用,因为计数它是无用的。但是,treeFold 需要才能有一个函数,它接受两个参数,所以你给它x。这是唯一的原因。你可以传递任何其他价值。