我正在使用python中的二叉树。我需要创建一个搜索树的方法,并返回一个新值可以插入的最佳节点。但是我有麻烦从这个递归函数返回一个值。我是python的新手。
def return_key(self, val, node):
if(val < node.v):
if(node.l != None):
self.return_key(val, node.l)
else:
print node.v
return node
else:
if(node.r != None):
#print node.v
self.return_key(val, node.r)
else:
print node.v
return node
打印node.v
打印节点值,但是当我打印返回的节点时:
print ((tree.return_key(6, tree.getRoot().v)))
它打印
没有结果是
。
您需要返回递归调用的结果。你在这里忽略了它:
if(node.l != None):
self.return_key(val, node.l)
和
if(node.r != None):
self.return_key(val, node.r)
递归调用与其他函数调用没有什么不同,如果有返回值,仍然需要处理返回值。使用return
语句:
if(node.l != None):
return self.return_key(val, node.l)
# ...
if(node.r != None):
return self.return_key(val, node.r)
注意,由于None
是一个单例值,您可以并且应该在这里使用is not None
来测试是否不存在:
if node.l is not None:
return self.return_key(val, node.l)
# ...
if node.r is not None:
return self.return_key(val, node.r)
我怀疑你给调用传递了错误的参数。如果第二个参数是一个节点,不要传入节点值:
print(tree.return_key(6, tree.getRoot())) # drop the .v
同样,如果你所有的node
类都有相同的方法,你可以递归到它,而不是使用self.return_value()
;在Tree
上只做:
print tree.return_key(6)
其中Tree.return_key()
委托给根节点:
def return_key(self, val):
root = tree.getRoot()
if root is not None:
return tree.getRoot().return_key(val)
and Node.return_key()
变为:
def return_key(self, val):
if val < self.v:
if self.l is not None:
return self.l.return_key(val)
elif val > self.v:
if self.r is not None:
return self.r.return_key(val)
# val == self.v or child node is None
return self
我在这里也更新了val
测试逻辑;如果val < self.v
(或代码中的val < node.v
)为假,不要假设val > self.v
为真;val
可以相等