Python二叉树

时间:2014-12-07 13:17:03

标签: python class binary-tree

我已经实现了一个简单的树,我遇到了这个问题。当我尝试搜索节点并且它存在时,即使它在查找方法中的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

1 个答案:

答案 0 :(得分:3)

当根目录中不存在该项时,您调用其子lookup()函数而不返回其值,因此即使代码在树中的某处找到数据,您也会获得None值而不是结果(1 / -1

替换这一行:

self.left.lookup(data)
...
self.right.lookup(data)

行:

return self.left.lookup(data)
...
return self.right.lookup(data)