我已经实现了一个简单的树,我遇到了这个问题。当我尝试搜索节点并且它存在时,即使它在查找方法中的if语句处运行print,它也会返回None。当我查找根节点时它运行正常。所有其余节点都返回None。有人能解释一下原因吗?
class Node():
def __init__(self,data):
self.right = None
self.left = None
self.data = data
def insert(self,data):
if self.data == data:
print "this item exists"
elif self.data > data:
if self.left == None:
self.left = Node(data)
else:
self.left.insert(data)
else:
if self.right == None:
self.right = Node(data)
else:
self.right.insert(data)
def print_nodes(self):
if self.left:
self.left.print_nodes()
print self.data
if self.right:
self.right.print_nodes()
def lookup(self,data):
if self.data == data:
print 'exists'
return 1
elif self.data > data:
if self.left != None:
self.left.lookup(data)
else:
return -1
elif self.data < data:
if self.right != None:
self.right.lookup(data)
else:
return -1
def delete(self,data):
if self.lookup(data)== -1:
print "doesnot exists"
else:
if (not self.left) and (not self.right):
self.data = None
root = Node(5)
#root.insert(3)
root.insert(3)
root.insert(2)
root.insert(6)
root.insert(61)
root.insert(62)
root.insert(63)
x = root.lookup(3)
print x
答案 0 :(得分:3)
当根目录中不存在该项时,您调用其子lookup()
函数而不返回其值,因此即使代码在树中的某处找到数据,您也会获得None
值而不是结果(1
/ -1
)
替换这一行:
self.left.lookup(data)
...
self.right.lookup(data)
行:
return self.left.lookup(data)
...
return self.right.lookup(data)