我的isElement函数(二叉树)出了什么问题?

时间:2015-06-03 12:50:27

标签: haskell binary-tree

我正在研究一个检查元素是否是二叉树的一部分的函数。我为我的树定义了一个名为Tree的类型,用于获取根元素和左右子树的函数,以及一个名为isElement的函数,用于检查值是否在我的树中。不幸的是,该函数仅适用于根元素。

以下示例说明了我从isElement函数获得的错误结果:

*Main>let tree = Node 1 Empty (Node 2 Empty (Node 3 Empty Empty))
*Main> isElement  tree 2
False
*Main> isElement  tree 3
False
*Main> isElement  tree 1
True

这是我的代码:

data Tree a = Node a (Tree a) (Tree a) 
        |Empty
        deriving (Show)

nodeValue :: Tree t -> t
nodeValue (Node x _ _) = x

rightTree :: Tree t -> Tree t
rightTree (Node _ _ x) = x

leftTree :: Tree t -> Tree t
leftTree (Node _ x _) = x

isNode ::  Tree t -> Bool 
isNode (Node _ _ _) = True
isNode _ = False

isElement :: (Eq t) => Tree t -> t -> Bool
isElement tree t
    |isNode tree == False = False 
    | nodeValue(tree) == t = True
    | otherwise = (isElement (leftTree(tree)) t) || (isElement (leftTree(tree)) t)

1 个答案:

答案 0 :(得分:8)

您的return acosl(dot(vec1, vec2) / (vec1.norm() * vec2.norm()));函数的定义存在错误:您正在调用isElement两次,而不是同时调用leftTreeleftTree。因此,从未探索过正确的子树。相应地修改代码,

rightTree

然后isElement tree t |isNode tree == False = False | nodeValue(tree) == t = True | otherwise = (isElement (leftTree(tree)) t) || (isElement (rightTree(tree)) 按宣传方式工作:

isElement

但是,仍有改进的余地。您并不真正需要所有这些功能(λ> let tree = Node 1 Empty (Node 2 Empty (Node 3 Empty Empty)) λ> isElement tree 2 True λ> isElement tree 3 True λ> isElement tree 1 True isNode等)来定义nodeValue。相反,您可以通过将模式匹配解压缩,将定义分解为两个等式:一个对应于树为空的情况,另一个对应于树为节点的情况:

isElement

编辑:正如路易斯·卡西利亚斯在his comment中指出的那样,这个替代定义的另一个好处是,它可以更好地(比原始定义)更好地进行穷举检查由编译器。