我正在研究一个检查元素是否是二叉树的一部分的函数。我为我的树定义了一个名为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)
答案 0 :(得分:8)
您的return acosl(dot(vec1, vec2) / (vec1.norm() * vec2.norm()));
函数的定义存在错误:您正在调用isElement
两次,而不是同时调用leftTree
和leftTree
。因此,从未探索过正确的子树。相应地修改代码,
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中指出的那样,这个替代定义的另一个好处是,它可以更好地(比原始定义)更好地进行穷举检查由编译器。