如何运行BST树功能的高度

debugcn 发表于 Dev

Surinder Kumar

有人可以帮我吗？我在python中实现了二进制搜索树数据结构，并编写了BST_height（）函数来计算树的高度。但是，当我运行我的代码时，它给我一个错误，提示“未定义自我”。我知道为什么会出现错误，但是您能建议其他方法来用根节点运行BST_height函数吗？

class Node:

    def __init__(self, data=None):
        self.data = data
        self.left = None
        self.right = None 

class BinarySearchTree:

    def __init__(self):
        self.root = Node()

    def display(self):

        print('''
                {}
              /   \\
             {}    {}
            / \\   / \\
          {}   {} {} {}  

           BINARY TREE         
        '''.format(tree.root.data, tree.root.left.data, tree.root.right.data, tree.root.left.left.data, tree.root.right.right.data, tree.root.left.right.data, tree.root.right.left.data))

    def checkRoot(self):

        if self.root.data != None:
            return 'Root node exists'
        else:
            return 'Root node doesn\'t exists'    

    def insert(self, data):
        newNode = Node(data)

        if self.root.data == None:
            # creating the root node 
            self.root = newNode
        else:
            self.insertNode(data, self.root)

    def insertNode(self, data, curNode):

        if data < curNode.data:
            if curNode.left == None:
                curNode.left = Node(data)
            else:
                self.insertNode(data, curNode.left)

        elif data > curNode.data:
            if curNode.right == None:
                curNode.right = Node(data)
            else:
                self.insertNode(data, curNode.right)

        else:
            print('The value already exists ha ha')# funny 

    def BST_height(self, node):                     

        if node == None:
            return -1

        leftHeight = height(node.left)
        rightHeight = height(node.right)

        return max(leftHeight, rightHeight) + 1 



tree = BinarySearchTree()
tree.insert(30)# root node 
tree.insert(24)
tree.insert(45)
tree.insert(90)
tree.insert(18)
tree.insert(28)
tree.insert(40)
tree.display()

# getting an error here
# I know self.root can\'t be used outside the class but can you suggest some other way tree.BST_height(self.root)

朋友

该代码有2个问题：

自我是您在类函数中使用的东西。在外面，您可以简单地使用object变量。像这样：

tree.BST_height(tree.root)

BST_height函数中有一个小错误。它调用height而不是self.BST_height：

def BST_height(self, node):
    if node == None:
        return -1

    leftHeight = self.BST_height(node.left)
    rightHeight = self.BST_height(node.right)

    return max(leftHeight, rightHeight) + 1

您可能需要阅读此内容，以使整个self主题更加清晰。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。