[英]How to return value from recursive function in python?
我正在使用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)
并且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
可以等于。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.